NOIP最优贸易

#include<cstdio>
#define MXN 100000+1
#define MXM 10*MXN
int getint(){
    int x=0,w=1;
    char c=getchar();
    while(c<‘0‘||c>‘9‘){
        if(c==‘-‘) w=0;
        c=getchar();
    }
    while(c>=‘0‘&&c<=‘9‘){
        x=(x<<3)+(x<<1)+(c-‘0‘);
        c=getchar();
    }
    return w?x:-x;
}
void write(int x){
    if(x<0){
        putchar(‘-‘);
        x=-x;
    }
    if(x>9) write(x/10);
    putchar(x%10+‘0‘);
    return;
}
int n,m,x,y,z,p,end;
int price[MXN],dis[3*MXN+2];
bool vis[3*MXN+2];
int queue[MXM],head,tail;
int u[3*MXM+2],v[3*MXM+2],w[3*MXM+2],fst[3*MXN+2],nxt[3*MXM+2],top;
void add(int x,int y,int z){
    top++;
    u[top]=x,v[top]=y,w[top]=z;
    nxt[top]=fst[u[top]];
    fst[u[top]]=top;
    return;
}
void spfa(){
    for(int i=2;i<=3*n+1;i++) dis[i]=2000000000;
    dis[1]=0;
    vis[1]=true;
    queue[tail++]=1;
    int now;
    while(head!=tail){
        now=queue[head];
        vis[now]=false;
        for(int i=fst[now];i;i=nxt[i]){
            if(dis[v[i]]>dis[now]+w[i]){
                dis[v[i]]=dis[now]+w[i];
                if(vis[v[i]]==false){
                    queue[tail]=v[i];
                    vis[v[i]]=true;
                    tail=tail+1==MXM ? 0 : tail+1;
                }
            }
        }
        head=head+1==MXM ? 0 : head+1;
    }
    return;
}
int main(){
    n=getint(),m=getint();
    for(int i=1;i<=n;i++) price[i]=getint();
    for(int i=0;i<m;i++){
        x=getint(),y=getint(),p=getint();
        if(p==1){
            add(x,y,0),add(x+n,y+n,0),add(x+2*n,y+2*n,0);
            add(x,y+n,price[x]),add(x+n,y+2*n,-price[x]);
        }
        if(p==2){
            add(x,y,0),add(y,x,0);
            add(x+n,y+n,0),add(y+n,x+n,0);
            add(x+2*n,y+2*n,0),add(y+2*n,x+2*n,0);
            add(x,y+n,price[x]),add(y,x+n,price[y]);
            add(x+n,y+2*n,-price[x]),add(y+n,x+2*n,-price[y]);
        }
    }
    end=3*n+1;
    add(n,end,0),add(3*n,end,0);
    spfa();
    write(-dis[end]);
    return 0;
}