POJ1062——Dijkstra算法的简单应用

年轻的探险家来到了一个印第安部落里。在那里他和酋长的女儿相爱了,于是便向酋长去求亲。酋长要他用10000个金币作为聘礼才答应把女儿嫁给他。探险家拿不出这么多金币,便请求酋长降低要求。酋长说:"嗯,如果你能够替我弄到大祭司的皮袄,我可以只要8000金币。如果你能够弄来他的水晶球,那么只要5000金币就行了。"探险家就跑到大祭司那里,向他要求皮袄或水晶球,大祭司要他用金币来换,或者替他弄来其他的东西,他可以降低价格。探险家于是又跑到其他地方,其他人也提出了类似的要求,或者直接用金币换,或者找到其他东西就可以降低价格。不过探险家没必要用多样东西去换一样东西,因为不会得到更低的价格。探险家现在很需要你的帮忙,让他用最少的金币娶到自己的心上人。另外他要告诉你的是,在这个部落里,等级观念十分森严。地位差距超过一定限制的两个人之间不会进行任何形式的直接接触,包括交易。他是一个外来人,所以可以不受这些限制。但是如果他和某个地位较低的人进行了交易,地位较高的的人不会再和他交易,他们认为这样等于是间接接触,反过来也一样。因此你需要在考虑所有的情况以后给他提供一个最好的方案。 
为了方便起见,我们把所有的物品从1开始进行编号,酋长的允诺也看作一个物品,并且编号总是1。每个物品都有对应的价格P,主人的地位等级L,以及一系列的替代品Ti和该替代品所对应的"优惠"Vi。如果两人地位等级差距超过了M,就不能"间接交易"。你必须根据这些数据来计算出探险家最少需要多少金币才能娶到酋长的女儿。 
Input
输入第一行是两个整数M,N(1 <= N <= 100),依次表示地位等级差距限制和物品的总数。接下来按照编号从小到大依次给出了N个物品的描述。每个物品的描述开头是三个非负整数P、L、X(X < N),依次表示该物品的价格、主人的地位等级和替代品总数。接下来X行每行包括两个整数T和V,分别表示替代品的编号和"优惠价格"。
Output
输出最少需要的金币数。
Sample Input
1 4
10000 3 2
2 8000
3 5000
1000 2 1
4 200
3000 2 1
4 200
50 2 0
Sample Output
5250


主要用到图最短路径Dijkstra算法,其中根据酋长等级来确定各个等级区间,根据等级区间来枚举,即分成多个子图,对每个子图用Dijkstra,最后取所有子图的Dijkstra的最小值。

代码:

#include<iostream>
#include<fstream>
#include<algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;
struct dingD{
	int index;
	int level;
};
int visited[150];
int limit[150];
int arc[150][150];
int dis[150];
dingD v[150];

int abs(int a,int b){
	if(a>b)
		return a-b;
	return b-a;
}

int Dij(int num){
	int result = INF;
	memset(visited,0,sizeof(visited));
	visited[0] = 1;
	int min;
	int jj;
	for(int i = 1;i<=num;i++){
		min = INF;
		for(int j = 1;j<=num;j++){
			if(!visited[j] && dis[j]<INF && limit[j]){
				jj = j;
				min = dis[j];
			}
		}
		visited[jj] = 1;

		for(int k = 1;k<=num;k++){
			if(limit[k] && dis[k] > dis[jj] + arc[jj][k]){
				dis[k] = dis[jj] + arc[jj][k];
			}
		}
	}

	return dis[1];
}

int main(){
	
	for(int i = 0;i<150;i++){
		dis[i] = INF;
		for(int j =0;j<150;j++){
			if(i==j){
				arc[i][j] = 0;
			}
			else{
				arc[i][j] = INF;
			}
		}
	}
	dis[0] = 0;
	memset(limit,0,sizeof(limit));

	fstream f;
	f.open("testData.txt",ios::in);
	
	int M,N;
	f>>M>>N;
	int num = N;
	while(num--){
		v[N-num].index = N-num;
		int P,L,X;
		f>>P>>L>>X;
		v[N-num].level = L;
		arc[0][N-num] = P;
		dis[N-num] = P;
		int xx = X;
		while(xx--){
			int T,V;
			f>>T>>V;
			arc[T][N-num] = V;
		}
	}
	int result = arc[0][1];
	for(int i = 0;i<=M;i++){
		memset(limit,0,sizeof(limit));
		for(int j = 1;j<=N;j++){
			if(v[j].level >= v[1].level-M+i && v[j].level <= v[1].level+i )
				limit[j] = 1;
		}
			result = min(result,Dij(N));
	}
	cout<<result;
	return 0;
}

其中limit数组用来是否处于等级区间。


猜你喜欢

转载自blog.csdn.net/qq_41508508/article/details/81019005
今日推荐