codeforces990F Flow Control

最开始的想法：同一个双连通分量中随便找一条边传递这个点的全部流量，遍历整个分量就行了，而缩点之后的图是一棵树，每次选择树种度为1的点，因为他只有一条边用来传递流量，依次类推，流完连通分量树后，再流每个连通分量的内部。于是洋洋洒洒写了一堆，结果wa on test 7。于是看到网上题解，想通了只要流量总和为0，然后在这个连通图中找一棵树流就行了，我依然写的每次找出度为1的点，结果 wa on test3.。。。。为撒按dfs顺序从子节点流向父节点就不会错啊？实在是不知道错哪了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#define maxl 400010

using namespace std;

int n,m,cnt,ssum;
int s[maxl],du[maxl],ehead[maxl];
struct edge1
{
	int x,y,ans;
}edge[maxl];
struct ed
{
	int u,to,nxt;bool used;
}e[maxl<<1];
typedef pair<int,int> p;
map <p,int> flow;
queue <int> q;
bool flag;
bool in[maxl];

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0' || ch>'9') {if(ch=='-') f=-f;ch=getchar();}
	while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
	return x*f;
}

inline void add(int u,int v)
{
	e[++cnt].to=v;e[cnt].u=u;e[cnt].nxt=ehead[u];
	ehead[u]=cnt;e[cnt].used=false;
}

inline void prework()
{
	n=read();ssum=0;flag=true;
	for(int i=1;i<=n;i++)
	{
		s[i]=read();
		ssum+=s[i];
	}
	m=read();
	memset(ehead,0,sizeof(ehead));cnt=1;
	for(int i=1;i<=m;i++)
	{
		edge[i].x=read();edge[i].y=read();edge[i].ans=0;
		add(edge[i].x,edge[i].y);add(edge[i].y,edge[i].x);
	}
	if(ssum)
	{
		flag=false;
		return;
	}
}

inline void dfs(int u,int fa)
{
	int v;
	for(int i=ehead[u];i;i=e[i].nxt)
	{
		v=e[i].to;
		if(in[v]) continue;
		du[u]++;du[v]++;in[v]=true;
		e[i].used=true;
		dfs(v,u);
		p d=make_pair(min(u,v),max(u,v));
		if(u<v)
			flow[d]+=s[v];
		else
			flow[d]-=s[v];
		s[u]+=s[v];s[v]=0;
	}
}

inline void mainwork()
{
	if(!flag)
		return;
	flow.clear();
	memset(in,false,sizeof(in));
	in[1]=true;
	dfs(1,0);
	int u,v;
	for(int i=1;i<=m;i++)
	{
		u=edge[i].x;v=edge[i].y; 
		p d=make_pair(min(u,v),max(u,v));
		if(flow.count(d))
		{
			if(u<v)
				edge[i].ans=flow[d];
			else
				edge[i].ans=-flow[d];
		}
		else
			edge[i].ans=0;
	}
}

inline void print()
{
	if(flag)
	{
		printf("Possible\n");
		for(int i=1;i<=m;i++)
			printf("%d\n",edge[i].ans);
	}
	else
		printf("Impossible\n");
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}

--------update：为撒我之前那个先缩点的程序改成树的dfs顺序过了呀？每次找出度为1的点和dfs没有区别吧？

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#define maxl 400010

using namespace std;

int n,m,ind,cnt,fcnt,ff,tot,top,ssum;
int dfn[maxl],low[maxl],f[maxl],sta[maxl];
int s[maxl],fs[maxl],ft[maxl],du[maxl];
int ehead[maxl],fehead[maxl];
struct edge1
{
	int x,y,ans;
}edge[maxl];
struct ed
{
	int u,to,nxt;
}e[maxl<<1],fe[maxl<<1];
typedef pair<int,int> p;
map <p,int> flow;
set <p> edset;
queue <int> q;
bool flag;
bool in[maxl];

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0' || ch>'9') {if(ch=='-') f=-f;ch=getchar();}
	while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
	return x*f;
}

inline void add(int u,int v)
{
	e[++cnt].to=v;e[cnt].u=u;e[cnt].nxt=ehead[u];ehead[u]=cnt;
}

inline void tarjan(int u,int fa)
{
	int v;
	dfn[u]=++ind;low[u]=ind;sta[++top]=u;
	for(int i=ehead[u];i;i=e[i].nxt)
	{
		v=e[i].to;
		if(v==fa) continue;
		if(!dfn[v])
		{
			tarjan(v,u);
			if(low[v]<low[u])
				low[u]=low[v];
			if(low[v]>dfn[u])
			{
				ff++;
				do
				{
					f[sta[top]]=ff;ft[ff]=sta[top];
					fs[ff]+=s[sta[top]];
				}
				while(sta[top--]!=v);
			}
		}else
		if(low[u]>dfn[v])
			low[u]=dfn[v];
	}
}

inline void fadd(int u,int v)
{
	fe[++fcnt].to=v;fe[fcnt].u=u;fe[fcnt].nxt=fehead[f[u]];fehead[f[u]]=fcnt;
}

inline void prework()
{
	n=read();ssum=0;flag=true;
	for(int i=1;i<=n;i++)
	{
		s[i]=read();
		ssum+=s[i];
	}
	m=read();
	memset(ehead,0,sizeof(ehead));cnt=0;
	for(int i=1;i<=m;i++)
	{
		edge[i].x=read();edge[i].y=read();edge[i].ans=0;
		add(edge[i].x,edge[i].y);add(edge[i].y,edge[i].x);
	}
	if(ssum)
	{
		flag=false;
		return;
	}
	ff=0;top=0;
	memset(dfn,0,sizeof(dfn));
	memset(low,0,sizeof(low));
	memset(f,0,sizeof(f));
	memset(fs,0,sizeof(fs));
	memset(ft,0,sizeof(ft));
	tarjan(1,0);
	for(int i=1;i<=n;i++)
	if(f[i]==0)
		fs[0]+=s[i],ft[0]=i;
	for(int i=0;i<=ff;i++)
		du[i]=0;
	int x,y;
	edset.clear();fcnt=0;
	memset(fehead,0,sizeof(fehead));
	for(int u=1;u<=n;u++)
		for(int i=ehead[u];i;i=e[i].nxt)
		{
			x=f[u];y=f[e[i].to];
			if(x!=y)
			{
				p d=make_pair(min(x,y),max(x,y));
				if(edset.find(d)==edset.end())
				{
					du[x]++,du[y]++;edset.insert(d);
					fadd(u,e[i].to);fadd(e[i].to,u);
				}
			}
		}
}

inline void fdfs(int fu,int fa)
{
	int u,v,fv;
	for(int i=fehead[fu];i;i=fe[i].nxt)
	{
		u=fe[i].u;v=fe[i].to;fv=f[v];
		if(in[fv]) continue;
		in[fv]=true;
		fdfs(fv,fu);
		p d=make_pair(min(u,v),max(u,v));
		if(u<v)
			flow[d]+=fs[fv];
		else
			flow[d]-=fs[fv];
		s[u]+=fs[fv];s[v]-=fs[fv];
		fs[fu]+=fs[fv];fs[fv]=0;
	}
}

inline void dfs(int u,int fa)
{
	int v,fu,fv;
	for(int i=ehead[u];i;i=e[i].nxt)
	{
		v=e[i].to;fu=f[u];fv=f[v];
		if(in[v] || fu!=fv) continue;
		in[v]=true;
		dfs(v,u);
		p d=make_pair(min(u,v),max(u,v));
		if(u<v)
			flow[d]+=s[v];
		else
			flow[d]-=s[v];
		s[u]+=s[v];s[v]-=s[v];
	}
}

inline void mainwork()
{
	if(!flag)
		return;
	while(!q.empty())
		q.pop();
	for(int i=0;i<=ff;i++)
	{
		if(du[i]==1 || du[i]==0)
			q.push(i);
		in[i]=false; 
	}
	int u,v,fu,fv;
	flow.clear();
	in[0]=true;
	fdfs(0,-1); 
	memset(in,false,sizeof(in));
	for(int i=0;i<=ff;i++)
	{
		in[ft[i]]=true;
		dfs(ft[i],0); 
	}
	
	for(int i=1;i<=m;i++)
	{
		u=edge[i].x;v=edge[i].y; 
		p d=make_pair(min(u,v),max(u,v));
		if(flow.count(d))
		{
			if(u<v)
				edge[i].ans=flow[d];
			else
				edge[i].ans=-flow[d];
		}
		else
			edge[i].ans=0;
	}
}

inline void print()
{
	if(flag)
	{
		printf("Possible\n");
		for(int i=1;i<=m;i++)
			printf("%d\n",edge[i].ans);
	}
	else
		printf("Impossible\n");
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}