洛谷 P1073 [NOIP2009 提高组] 最优贸易 【分层图最短路】

题目链接

题意：

给出一个由无向边和有向边构成的 n 个结点的图，起点为1，终点为n，你可以在某个结点 $i$ 花费 $V_i$ 买入物品，再另一个结点 $j$ 卖出得到 $V_j$，这种操作只能进行一次，求最大利益

题解：

第一层是原图，第二层是买入后的图，第三层是卖出后的图，第一层 $i$ 连接到第二层 $i+n$ ，边权 $-V_i$，第二层 $i+n$ 连接到第三层 $i+2\ast n$ ，边权 $V_i$，求 1 到 3*n 的最长路就好了

#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<list>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
#include<random>
using namespace std;
#include<ext/pb_ds/priority_queue.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>
using namespace __gnu_pbds;
#include<ext/rope>
using namespace __gnu_cxx;

#define int long long
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))

const int mod = 1e9+7;
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return ans;
}
const int INF = 0x3f3f3f3f;
const int N = 1e7+10;
struct node{
	int to,next,val;
}e[N];
int head[N],cnt;
int v[N],dis[N],vis[N],n,m;
void add(int u,int v,int w){
	e[++cnt]={v,head[u],w}; head[u]=cnt;
}
void spfa(){
	for(int i=1;i<=3*n;i++)dis[i]=-INF;
	dis[1]=0;
	queue<int>q; q.push(1); vis[1]=1;
	while(!q.empty()){
		int u=q.front(); q.pop(); vis[u]=0;
		for(int i=head[u];i;i=e[i].next){
			int v=e[i].to,w=e[i].val;
			if(dis[v]<dis[u]+w){
				dis[v]=dis[u]+w;
				if(!vis[v])q.push(v),vis[v]=1;
			}
		}
	}
}
#define endl '\n'
signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>v[i];
	for(int i=1;i<=m;i++){
		int u,v,op; cin>>u>>v>>op;
		add(u,v,0); if(op==2)add(v,u,0);//第一层 
		u+=n,v+=n; 
		add(u,v,0); if(op==2)add(v,u,0);//第二层 
		u+=n,v+=n; 
		add(u,v,0); if(op==2)add(v,u,0);//第三层 
	}
	for(int i=1;i<=n;i++)add(i,i+n,-v[i]),add(i+n,i+2*n,v[i]);
	//连接1、2、3层
	spfa(); 
	cout<<dis[3*n]<<endl;
}