直接最大流。。。。由于只用找两条路径,spfa跑两次就可以了。
#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define MAXN 310000
#define MAXM 310000
#define INF 0x3f3f3f3f
struct node
{
int u,v,f,c,next;
}e[MAXM],tree[MAXM];
int k,head[MAXN],pre[MAXN],dist[MAXN],vis[MAXN];
int next[MAXN],first[MAXN],to[MAXN],cost[MAXN],flow[MAXN],fa[MAXN];
int en,s,t,maxflow,mincost,m,n,num; //s源点,t汇点
void add(int u,int v,int c,int f)
{
e[en].u=u;
e[en].v=v;
e[en].c=c;
e[en].f=f;
e[en].next=head[u];
head[u]=en++;
e[en].u=v;
e[en].v=u;
e[en].c=-c;
e[en].f=0;
e[en].next=head[v];
head[v]=en++;
}
int spfa()
{
int i,u,v;
for(i=0;i<=t;i++)
pre[i]=-1,vis[i]=0,dist[i]=INF;
dist[s]=0;
vis[s]=1;
queue<int>q;
q.push(s);
while(!q.empty())
{
u=q.front();
q.pop();
for(i=head[u];i!=-1;i=e[i].next)
{
v=e[i].v;
if(e[i].f>0&&dist[u]+e[i].c<dist[v])
{
dist[v]=dist[u]+e[i].c;
pre[v]=i;
if(!vis[v])
{
vis[v]=1;
q.push(v);
}
}
}
vis[u]=0;
}
if(dist[t]==INF)
return 0;
return 1;
}
void add2()
{
int v;
int maxf=INF;
for(v=pre[t];~v;v=pre[e[v].u])
maxf=min(maxf,e[v].f);
for(v=pre[t];~v;v=pre[e[v].u])
{
e[v].f-=maxf;
e[v^1].f+=maxf;
}
maxflow+=maxf;
mincost+=maxf*dist[t];
}
void init()
{
maxflow=0;
mincost=0;
s=1;
t=n*2;
memset(head,-1,sizeof(head));
}
int main()
{
int i,j,a,b,c,d;
while(~scanf("%d%d",&n,&m))
{
init();
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&a,&b,&c);
add(a+n,b,c,1);
}
for(int i=2;i<n;i++)
{
add(i,i+n,0,1);
}
add(1,1+n,0,10);
add(n,n+n,0,10);
for(int i=1;i<=2;i++)//只用找2条路径,所以只用2次
{
if(spfa())
add2();
}
printf("%d\n",mincost);
}
return 0;
}
/*
3 10 2 3 3 2
5
6
7
*/