Shoot the Bullet ZOJ

Code：

#include<cstdio>        
#include<algorithm>
#include<vector>
#include<queue>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn=2005;
const int maxd=1000000+233;
const int INF=10000000;
# define  pb push_back
int sss,ttt;
int s,t;
int E1,E2;
int mapp[maxd],A[maxn],val[maxd];
struct Edge{
	int from,to,cap;
	Edge(int u,int v,int c):from(u),to(v),cap(c) {}
};
vector<Edge>edges;
struct Dicnic{
	vector<int>G[maxn];
	int d[maxn],vis[maxn],cur[maxn];
	queue<int>Q;
	void init(){
		for(int i=0;i<maxn;++i)G[i].clear();
		edges.clear();
	}
	void addedge(int u,int v,int c,int cnt)
	{
		edges.pb(Edge(u,v,c));               
		edges.pb(Edge(v,u,0));               
		int m=edges.size();
		G[u].pb(m-2);                                   //正向弧
		G[v].pb(m-1);                                   //反向弧
		E1=m-2,E2=m-1;
		if(cnt>0)mapp[cnt]=m-1;
	}
	int BFS()
	{
		memset(vis,0,sizeof(vis));
		d[s]=0,vis[s]=1;Q.push(s);
		while(!Q.empty())
		{
			int u=Q.front();Q.pop();
			int sz=G[u].size();
			for(int i=0;i<sz;++i){
				Edge e=edges[G[u][i]];
				if(!vis[e.to]&&e.cap>0&&e.to!=sss&&e.to!=ttt){
					d[e.to]=d[u]+1,vis[e.to]=1;
					Q.push(e.to);
				}
			}
		}
		return vis[t];
	}
	int dfs(int x,int a)
	{
			if(x==t)return a;
			int sz=G[x].size();
			int f,flow=0;
			for(int i=cur[x];i<sz;++i)
			{
				Edge e=edges[G[x][i]];
				cur[x]=i;
				if(d[e.to]==d[x]+1&&e.cap>0&&e.to!=sss&&e.to!=ttt)
				{
					f=dfs(e.to,min(a,e.cap));
					if(f)
					{
						int u=G[x][i];
						a-=f;
						edges[u].cap-=f;
						edges[u^1].cap+=f;
						flow+=f;
						if(a==0)break;
					}
				}
			}
			return flow;
	}
	int maxflow(){
		int ans=0;
		while(BFS()){
		    memset(cur,0,sizeof(cur));
			ans+=dfs(s,INF);
		}
		return ans;
	}
}op;
int main()
{
	int n,m;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
	op.init();
	memset(A,0,sizeof(A));
	int sum=0,cnt=0,ss,tt;
	ss=0,tt=n+m+1;
	for(int i=0;i<m;++i){
		int a;scanf("%d",&a);
		op.addedge(n+1+i,tt,INF-a,0);
		A[n+1+i]-=a,A[tt]+=a;
	}
	for(int i=1;i<=n;++i){
		int c,d;scanf("%d%d",&c,&d);
		op.addedge(ss,i,d,0);
		for(int j=1;j<=c;++j)
		{
			int x,y,z;
			scanf("%d%d%d",&x,&y,&z);
			++cnt;
			op.addedge(i,x+n+1,z-y,cnt);
			val[cnt]=y;
			A[i]-=y,A[x+n+1]+=y;
		}
	}
	s=1500,t=1501;
	for(int i=0;i<=n+m+1;++i)
	{
		if(A[i]>0){op.addedge(s,i,A[i],0);sum+=A[i];}
		if(A[i]<0)op.addedge(i,t,-A[i],0);
	}
	op.addedge(tt,ss,INF,0);
	sss=-1,ttt=-2;
	int ans1=op.maxflow();
	sss=1500,ttt=1501;
	if(ans1!=sum){printf("-1
");printf("
");}
	else
	{
		ans1=edges[E2].cap;
		edges[E1].cap=edges[E2].cap=0;
		s=ss,t=tt;
		ans1+=op.maxflow();
		printf("%d
",ans1);
		for(int i=1;i<=cnt;++i)
		    printf("%d
",edges[mapp[i]].cap+val[i]);
		printf("
");
	}
	return 0;
}