最开始的想法:同一个双连通分量中随便找一条边传递这个点的全部流量,遍历整个分量就行了,而缩点之后的图是一棵树,每次选择树种度为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;
}