P1073 最优贸易

题面:https://www.luogu.org/problem/P1073

建三层图,下第一层表示买,下第二层表示卖,没下表示没有进行贸易,然后跑最长路即可.
Code:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
#include<ctime>
#include<queue>
#include<vector>
using namespace std;
const int N=1000005;
int n,m,s,tot,Next[N],head[N],to[N],c[N],dis[N],val[N];
bool vis[N];
void add(int x,int y,int z){
    tot++;
    to[tot]=y;
    val[tot]=z;
	Next[tot]=head[x];
	head[x]=tot;
}
void Push(int x,int y){
	add(x,y,0);
	add(x+n,y+n,0);
	add(x+2*n,y+2*n,0);
	add(x,y+n,-c[x]);
	add(x+n,y+2*n,c[x]);
}
void SPFA(){
	queue<int> Q;
	for(int i=1;i<=n;i++){
		dis[i]=-2147483647;
		vis[i]=0;
	}
	vis[s]=1;
	dis[s]=0;
	Q.push(s);
	while(!Q.empty()){
		int u=Q.front();
		Q.pop();
		vis[u]=0;
		for(int i=head[u];i;i=Next[i]){
			int v=to[i];
			if(dis[v]<dis[u]+val[i]){
				dis[v]=dis[u]+val[i];
				if(!vis[v]){
					vis[v]=1;
					Q.push(v);
				}
			}
		}
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d",&c[i]);
	}
	for(int i=1;i<=m;i++){
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		if(z==1){
			Push(x,y);
		}
		else{
			Push(x,y);
			Push(y,x);
		}
	}
	add(n,3*n+1,0);
	add(3*n,3*n+1,0);
	s=1;
	n=3*n+1;
	SPFA();
	printf("%d\n",dis[n]);
	return 0;
}