拆点费用流。
每个点拆成出点和入点,S向每个出点连边(v=1,c=0),代表从该点出发可以有一条指向其他点的边。S向每个入点连边(v=1,c=Ai),代表可以以此点为起点。每个入点向T连边(v=1,c=0),代表到达该点。对应边的起点的出点向终点的入点连边(v=1,c=w),代表可以从起点出发到达终点,由此,任意一个入点要么从S出发(代价为Ai)要么从另一点的出点出发(代价为w),符合题意。
然后跑一遍费用流就过了O(∩_∩)O
#include<iostream> #include<cstdio> #include<cstring> #include<queue> using namespace std; const int inf=1e9; struct Edge{int from,to,next,v,c;}e[50005]; int head[1605],d[1605],cnt=1,from[1605]; bool inq[1605]; void ins(int u,int v,int w,int c){ cnt++;e[cnt].from=u;e[cnt].to=v;e[cnt].next=head[u];head[u]=cnt;e[cnt].v=w;e[cnt].c=c; } void insert(int u,int v,int w,int c){ ins(u,v,w,c);ins(v,u,0,-c); } bool spfa(int s,int t,int &cost){ memset(d,0x3f,sizeof(d)); d[s]=0;queue<int>q;q.push(s); while(!q.empty()){ int u=q.front();q.pop();inq[u]=false; for(int i=head[u];i;i=e[i].next) if(e[i].v&&d[e[i].to]>d[u]+e[i].c){ d[e[i].to]=d[u]+e[i].c; from[e[i].to]=i; if(!inq[e[i].to]){inq[e[i].to]=true;q.push(e[i].to);} } } if(d[t]>=inf)return false; int x=inf; for(int i=from[t];i;i=from[e[i].from])x=min(x,e[i].v); for(int i=from[t];i;i=from[e[i].from]){ e[i].v-=x;e[i^1].v+=x;cost+=x*e[i].c; } return true; } int mcmf(int s,int t){int cost=0;while(spfa(s,t,cost));return cost;} int main(){ int n,m,u,v,w;scanf("%d%d",&n,&m); int S=0,T=n+n+1; for(int i=1;i<=n;i++){ scanf("%d",&w); insert(S,i,1,0); insert(S,i+n,1,w); insert(i+n,T,1,0); } while(m--){ scanf("%d%d%d",&u,&v,&w); if(u>v)swap(u,v); insert(u,v+n,1,w); } printf("%d",mcmf(S,T)); return 0; }