HDU 4411 Arrest

 1 #include<algorithm>
 2 #include<cstdio>
 3 #include<cmath>
 4 #include<cstring>
 5 #include<iostream>
 6 #define inf 1000000
 7 int a[505][505],K;
 8 int tot,go[200005],next[200005],first[200005],cost[200005],flow[200005];
 9 int op[200005],dis[200005],c[200005],vis[200005],edge[200005],from[200005];
10 int S,T,n,m,ans;
11 int read(){
12     char ch=getchar();int t=0,f=1;
13     while (ch<‘0‘||ch>‘9‘){if (ch==‘-‘) f=-1;ch=getchar();}
14     while (‘0‘<=ch&&ch<=‘9‘){t=t*10+ch-‘0‘;ch=getchar();}
15     return t*f;
16 }
17 void insert(int x,int y,int z,int l){
18     tot++;
19     go[tot]=y;
20     next[tot]=first[x];
21     first[x]=tot;
22     flow[tot]=z;
23     cost[tot]=l;
24 }
25 void add(int x,int y,int z,int l){
26     insert(x,y,z,l);op[tot]=tot+1;
27     insert(y,x,0,-l);op[tot]=tot-1;
28 }
29 bool spfa(){
30     for (int i=0;i<=T;i++)
31      dis[i]=0x3f3f3f3f,vis[i]=0;
32     int h=1,t=1;c[1]=S;vis[S]=1;dis[S]=0;
33     while (h<=t){
34         int now=c[h++];
35         for (int i=first[now];i;i=next[i]){
36             int pur=go[i];
37             if (dis[pur]>dis[now]+cost[i]&&flow[i]){
38                 dis[pur]=dis[now]+cost[i];
39                 edge[pur]=i;
40                 from[pur]=now;
41                 if (vis[pur]) continue;
42                 vis[pur]=1;
43                 c[++t]=pur;
44             }
45         }
46         vis[now]=0;
47     }
48     return dis[T]!=0x3f3f3f3f;
49 }
50 void updata(){
51     int mn=0x3f3f3f3f;
52     for (int i=T;i!=S;i=from[i]){
53         mn=std::min(mn,flow[edge[i]]);
54     }
55     for (int i=T;i!=S;i=from[i]){
56         ans+=mn*cost[edge[i]];
57         flow[edge[i]]-=mn;
58         flow[op[edge[i]]]+=mn;
59     }
60 }
61 int main(){
62     while (scanf("%d",&n)!=EOF){
63         m=read();K=read();
64         if (n==0&&K==0&&m==0) return 0;
65         for (int i=0;i<=T;i++) first[i]=0;tot=0;
66         S=2*n+1;T=S+1;
67         for (int i=0;i<=n;i++)
68          for (int j=0;j<=n;j++)
69           if (i==j) a[i][j]=0;
70           else a[i][j]=0x3f3f3f3f;
71         while(m--){
72             int u=read(),v=read(),w=read();
73             a[u][v]=std::min(a[u][v],w);
74             a[v][u]=std::min(a[v][u],w);
75         }
76         for (int k=0;k<=n;k++)
77          for (int i=0;i<=n;i++)
78           for (int j=0;j<=n;j++)
79            a[i][j]=std::min(a[i][j],a[i][k]+a[k][j]);
80         add(S,0,K,0);
81         add(0,T,K,0);
82         for (int i=1;i<=n;i++)
83          if (a[0][i]<0x3f3f3f3f)
84           add(0,i,inf,a[0][i]);
85         for (int i=1;i<=n;i++)
86          for (int j=i+1;j<=n;j++)
87           if (a[i][j]<0x3f3f3f3f)
88            add(i+n,j,1,a[i][j]);
89         for (int i=1;i<=n;i++)
90           if (a[0][i]<0x3f3f3f3f)
91            add(i+n,T,1,a[0][i]);
92         for (int i=1;i<=n;i++)
93          add(i,i+n,1,-inf);      
94         ans=inf*n;
95         while (spfa()) updata();
96         printf("%d\n",ans);  
97     }
98 }