bzoj3258 秘密任务（Dij+最小割+tarjan）

首先跑两边Dijkstra，只留下一定存在在最短路上的边。
然后把一条边拆成两条边，权值分别为起终点的权值。
现在我们就是要花最小的代价把1,n割开，于是求一个最小割。
还要判定最小割是否唯一。可以对残余网络tarjan缩点。
如果一条边是割边，那么需要满足：
1、满流边
2、bel[x]!=bel[y]
如果还满足bel[x]==bel[s]&&bel[y]==bel[T]，那么这条割边一定会被割
否则不一定，最小割也就不唯一了。

#include <bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
#define inf 0x3f3f3f3f
#define N 5010
#define pa pair<ll,int>
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(T==S){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,a[410],h[N],num=1,h1[N],num1=0,lev[N],cur[N];
int dfn[N],low[N],dfnum,scc,bel[N];
ll dis[2][410];
bool inq[N];
struct edge{
    int to,next,val;
}data1[N<<1],data[N<<2];
inline void add(int x,int y,int val){
    data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=val;
    data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=0;
}
inline void Dij(int id){
    priority_queue<pa,vector<pa>,greater<pa> >q;
    memset(dis[id],inf,sizeof(dis[id]));
    if(id) q.push(make_pair(0,n)),dis[id][n]=0;
    else q.push(make_pair(0,1)),dis[id][1]=0;
    while(!q.empty()){
        int x=q.top().second;q.pop();
        if(inq[x]) continue;inq[x]=1;
        for(int i=h1[x];i;i=data1[i].next){
            int y=data1[i].to;
            if(dis[id][x]+data1[i].val<dis[id][y]){
                dis[id][y]=dis[id][x]+data1[i].val;
                q.push(make_pair(dis[id][y],y));
            }
        }
    }
}
inline bool bfs(){
    queue<int>q;memset(lev,0,sizeof(lev));
    lev[1]=1;q.push(1);
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;if(lev[y]||!data[i].val) continue;
            lev[y]=lev[x]+1;if(y==n) return 1;q.push(y);
        }
    }return 0;
}
inline int dinic(int x,int low){
    if(x==n) return low;int tmp=low;
    for(int &i=cur[x];i;i=data[i].next){
        int y=data[i].to;if(lev[y]!=lev[x]+1||!data[i].val) continue;
        int res=dinic(y,min(tmp,data[i].val));
        if(!res) lev[y]=0;else tmp-=res,data[i].val-=res,data[i^1].val+=res;
        if(!tmp) return low;
    }return low-tmp;
}stack<int>qq;
inline void tarjan(int x){
    dfn[x]=low[x]=++dfnum;qq.push(x);inq[x]=1;
    for(int i=h[x];i;i=data[i].next){
        if(!data[i].val) continue;int y=data[i].to;
        if(!dfn[y]) tarjan(y),low[x]=min(low[x],low[y]);
        else if(inq[y]) low[x]=min(low[x],dfn[y]);
    }if(low[x]==dfn[x]){
        ++scc;while(1){
            int y=qq.top();qq.pop();inq[y]=0;
            bel[y]=scc;if(y==x) break;
        }
    }
}
int main(){
//  freopen("a.in","r",stdin);
    int tst=read();
    while(tst--){
        n=read();m=read();memset(h1,0,sizeof(h1));num1=0;
        for(int i=1;i<n;++i) a[i]=read();a[n]=inf;
        while(m--){
            int x=read(),y=read(),val=read();
            data1[++num1].to=y;data1[num1].next=h1[x];h1[x]=num1;data1[num1].val=val;
            data1[++num1].to=x;data1[num1].next=h1[y];h1[y]=num1;data1[num1].val=val;
        }Dij(0);memset(inq,0,sizeof(inq));
        Dij(1);memset(inq,0,sizeof(inq));
        m=n;memset(h,0,sizeof(h));num=1;
        for(int x=1;x<n;++x)
            for(int i=h1[x];i;i=data1[i].next){
                int y=data1[i].to;
                if(dis[0][x]+data1[i].val+dis[1][y]!=dis[0][n]) continue;
                ++m;add(x,m,a[x]);add(m,y,a[y]);
            }ll ans=0;
        while(bfs()){memcpy(cur,h,sizeof(h));ans+=dinic(1,inf);}
        memset(dfn,0,sizeof(dfn));dfnum=0;scc=0;bool flag=0;
        for(int i=1;i<=m;++i) if(!dfn[i]) tarjan(i);
        for(int i=2;i<=num;i+=2){
            if(data[i].val) continue;
            int x=data[i^1].to,y=data[i].to;
            if(bel[x]==bel[y]) continue;
            if(bel[x]!=bel[1]||bel[y]!=bel[n]) flag=1;
        }if(flag) printf("No %lld\n",ans);
        else printf("Yes %lld\n",ans);
    }return 0;
}