记录编号 285025 评测结果 AAAAAAAAAA
题目名称 [网络流24题] 太空飞行计划 最终得分 100
用户昵称 GravatarTenderRun 是否通过 通过
代码语言 C++ 运行时间 0.011 s
提交时间 2016-07-20 16:07:25 内存使用 13.95 MiB
显示代码纯文本
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
using namespace std;
const int maxn=100010;
const int maxm=1000010;
const int INF=1000000000;
int cnt,fir[maxn],to[maxm],nxt[maxm],cap[maxm];
int dis[maxn],gap[maxn],path[maxn],fron[maxn];
bool a[maxn],b[maxn];
queue<int>q;
int n,m,S,T,tot=0;
struct Net_Flow{
	int tot;
	void Init(int tot_){
		memset(dis,0,sizeof(dis));
		memset(gap,0,sizeof(gap));
		memset(fir,0,sizeof(fir));
		cnt=1;tot=tot_;
	}
	void add(int a,int b,int c){
		nxt[++cnt]=fir[a];
		cap[cnt]=c;
		fir[a]=cnt;
		to[cnt]=b;
	}
	void addedge(int a,int b,int c){
		add(a,b,c);add(b,a,0);
	}
	bool BFS(int S,int T){
		dis[T]=1;q.push(T);
		while(!q.empty()){
			int x=q.front();q.pop();
			for(int i=fir[x];i;i=nxt[i])
				if(!dis[to[i]]){
					dis[to[i]]=dis[x]+1;
					q.push(to[i]);
				}
		}
		return dis[S];
	}
	int Max_Flow(int S,int T){
		if(!BFS(S,T))return 0;
		for(int i=0;i<tot;i++)fron[i]=fir[i];
		for(int i=0;i<tot;i++)gap[dis[i]]+=1;
		int ret=0,p=S,f,Min;
		while(dis[S]<=tot){
			if(p==T){
				f=INF;
				while(p!=S){
					f=min(f,cap[path[p]]);
					p=to[path[p]^1];
				}ret+=f;p=T;
				while(p!=S){
					cap[path[p]]-=f;
					cap[path[p]^1]+=f;
					p=to[path[p]^1];
				}
			}
			
			for(int &i=fron[p];i;i=nxt[i])
				if(cap[i]&&dis[to[i]]==dis[p]-1)
					{path[p=to[i]]=i;break;}
			
			if(!fron[p]){Min=tot;
				if(--gap[dis[p]]==0)break;
				for(int i=fir[p];i;i=nxt[i])
					if(cap[i])Min=min(Min,dis[to[i]]);
				gap[dis[p]=Min+1]+=1;fron[p]=fir[p];
				if(p!=S)p=to[path[p]^1];	
			}		
		}
		return ret;
	}
	bool vis[maxn];
	void DFS(int p){
		vis[p]=1;
		if(p<=n)a[p]=1;else b[p]=1;
		for(int i=fir[p];i;i=nxt[i])
			if(cap[i]&&!vis[to[i]])DFS(to[i]);
	}
	int Solve(int S,int T){
		int ret=Max_Flow(S,T);DFS(S);
		for(int i=n+1;i<=n+m;i++)
			if(b[i])printf("%d ",i-n);
		printf("\n");	
		for(int i=1;i<=n;i++)
			if(a[i])printf("%d ",i);
		printf("\n");
		return ret;			
	}
}ISAP;

int main(){
	freopen("shuttle.in","r",stdin);
	freopen("shuttle.out","w",stdout);
	scanf("%d%d",&m,&n);
	S=0;T=n+m+1;ISAP.Init(T+1);
	for(int i=n+1;i<=n+m;i++){
		int v,a;char c;
		scanf("%d",&v);tot+=v;
		ISAP.addedge(S,i,v);
		c=getchar();
		while(c!='\r'){
			scanf("%d",&a);
			ISAP.addedge(i,a,INF);
			c=getchar();
		}
	}
	for(int i=1,v;i<=n;i++){
		scanf("%d",&v);
		ISAP.addedge(i,T,v);
	}
	printf("%d\n",tot-ISAP.Solve(S,T));
	return 0;
}