P1073 最优贸易

Posted 2021-03-25 ukcxrtjr

题面: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
",dis[n]);
    return 0;
}