C++圖論SPFA算法例題———————最優貿易

 

題目描述:

C 國有

n個大城市和m 條道路,每條道路連接這n 個城市中的某兩個城市。任意兩個城市之間最多隻有一條道路直接相連。這m 條道路中有一部分爲單向通行的道路,一部分爲雙向通行的道路,雙向通行的道路在統計條數時也計爲1 條。C 國幅員遼闊,各地的資源分佈情況各不相同,這就導致了同一種商品在不同城市的價格不一定相同。但是,同一種商品在同一個城市的買入價和賣出價始終是相同的。商人阿龍來到 C 國旅遊。當他得知同一種商品在不同城市的價格可能會不同這一信息之後,便決定在旅遊的同時,利用商品在不同城市中的差價賺回一點旅費。設 C 國n 個城市的標號從1\sim n ,阿龍決定從1 號城市出發,並最終在n 號城市結束自己的旅行。在旅遊的過程中,任何城市可以重複經過多次,但不要求經過所有n 個城市。阿龍通過這樣的貿易方式賺取旅費:他會選擇一個經過的城市買入他最喜歡的商品——水晶球,並在之後經過的另一個城市賣出這個水晶球,用賺取的差價當做旅費。由於阿龍主要是來 C 國旅遊,他決定這個貿易只進行最多一次,當然,在賺不到差價的情況下他就無需進行貿易。

假設 C 國有5 個大城市,城市的編號和道路連接情況如下圖,單向箭頭表示這條道路爲單向通行,雙向箭頭表示這條道路爲雙向通行。

27.png

假設

1~n 號城市的水晶球價格分別爲4,3,5,6,1 。

阿龍可以選擇如下一條線路:

1, 2, 3, 5 ,並在2 號城市以3 的價格買入水晶球,在3 號城市以5 的價格賣出水晶球,賺取的旅費數爲2 。

阿龍也可以選擇如下一條線路1, 4, 5, 4, 5 ,並在第1 次到達5 號城市時以1 的價格買入水晶球,在第2 次到達4 號城市時以6 的價格賣出水晶球,賺取的旅費數爲5 。

現在給出n 個城市的水晶球價格,m 條道路的信息(每條道路所連接的兩個城市的編號以及該條道路的通行情況)。請你告訴阿龍,他最多能賺取多少旅費。

輸入:

輸入第一行包含

2個正整數n 和m ,中間用一個空格隔開,分別表示城市的數目和道路的數目。

第二行n 個正整數,每兩個整數之間用一個空格隔開,按標號順序分別表示這n 個城市的商品價格。

接下來m 行,每行有3 個正整數,x,y,z ,每兩個整數之間用一個空格隔開。如果z=1 ,表示這條道路是城市x 到城市y 之間的單向道路;如果z=2 ,表示這條道路爲城市x 和城市y 之間的雙向道路。

輸出:

輸出共1行,包含1 個整數,表示最多能賺取的旅費。如果沒有進行貿易,則輸出0 。

輸入樣例:

5 5
4 3 5 6 1
1 2 1
1 4 1
2 3 2
3 5 1
4 5 2

輸出樣例:

5

 思路分析:

這一題通過思考就是求一條路徑上每一點所能達到的最小購入值與最大的賣出值的差值。

但是我們該如何去求呢?

很簡單,用一次SPFA用正圖,求出每一點所能到的最小值(畢竟阿福是從1號節點開始的)。

那麼,再用一遍SPFA用原圖的逆圖,就能求出每一點所能到的最大賣出值了。(反向操作,最爲致命)。

最後再遍歷了一遍每個點, 就有答案了。

代碼實現:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int MAXN=100005;
const int MAXM=500005;
const int INF=100000;
int N,M,np=0,np2=0,last[MAXN],last2[MAXN],MAX[MAXN],MIN[MAXN],in[MAXN],a[MAXN];
struct edge{
    int to,pre;
};
edge E[MAXM*2],G1[MAXM*2];
char c;
void read(int &x)
{
	for(c=getchar();c<'0'||c>'9';c=getchar());
	for(x=0;c>='0'&&c<='9';c=getchar())
        x=x*10+c-'0';
}

void addedge(int u,int v)
{
	E[++np]=(edge){v,last[u]};
	last[u]=np;
}

void addedgG1(int u,int v)
{
	G1[++np2]=(edge){v,last2[u]};
	last2[u]=np2;
}

void spfa1()
{
	queue<int>q;
	for(int i=1;i<=N;i++) MIN[i]=INF;
	MIN[1]=a[1]; q.push(1);
	while(!q.empty())
	{
		int i=q.front(); q.pop();
		in[i]=0;
		for(int p=last[i];p;p=E[p].pre)
		{
			int j=E[p].to;
			if(MIN[j]==INF||MIN[j]>MIN[i])
			{
				MIN[j]=min(a[j],MIN[i]);
				if(!in[j])
				{
					in[j]=1;
					q.push(j);
				}
			}
		}
	}
}

void spfa2()
{
	queue<int>q;
	memset(in,0,sizeof(in));
	for(int i=1;i<=N;i++)
        MAX[i]=-1;
	MAX[N]=a[N];
	q.push(N);
	while(!q.empty())
	{
		int i=q.front(); q.pop(); in[i]=0;
		for(int p=last2[i];p;p=G1[p].pre)
		{
			int j=G1[p].to;
			if(MAX[j]==-1||MAX[j]<MAX[i])
			{
				MAX[j]=max(a[j],MAX[i]);
				if(!in[j])
				{
					in[j]=1;
					q.push(j);
				}
			}
		}
	}
}

int main()
{
	int i,u,v,op;
	read(N);
	read(M);
	for(i=1;i<=N;i++)
        read(a[i]);
	for(i=1;i<=M;i++)
	{
		read(u);
		read(v);
		read(op);
		if(op==2)
            addedge(v,u),addedgG1(u,v);
		addedge(u,v);addedgG1(v,u);
	}
	spfa1();
	spfa2();
	int ans=0;
	for(i=1;i<=N;i++)
        ans=max(ans,MAX[i]-MIN[i]);
	printf("%d",ans);
}

 

發佈了50 篇原創文章 · 獲贊 53 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章