codeforce 990F

题目链接:点击打开链接

给定每一个点的初始存量,通过分配管道的流量使得每个点最后的存量为0

看了大神的题解,如果所有点的初始存量和不为0,那么直接输出impossible。(因为管道无论怎么分配都不会影响存量总量)

然后生成一棵树,进行dfs。到达叶节点时处理之后回溯。每次对当前节点和其父节点的边进行处理,将当前节点的存量变为0,然后更新父节点的存量。

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define LL long long
#define mod 1000000007
#define inf 0x3f3f3f3f
#define sqr(a) (a)*(a)
#define For(i,m,n) for(int i=m;i<=n;i++)
#define Dor(i,n,m) for(int i=n;i>=m;i--)
#define lan(a,b) memset(a,b,sizeof(a))
#define maxn 200010

using namespace std;

int n,m;
LL s[maxn];
vector> edge[maxn];
LL ans[maxn];
int vis[maxn];

void dfs(int x,int fa,int id)
{
    vis[x]=1;
    for(int i=0;i0)
            ans[id]-=s[x];
        else
            ans[-id]+=s[x];
        s[fa]+=s[x];
        s[x]=0;
    }
}

int main()
{
    while(~scanf("%d",&n))
        {
            lan(ans,0);
            lan(s,0);
            lan(vis,0);
            LL sum=0;
            For(i,1,n)
            scanf("%lld",&s[i]),sum+=s[i],s[i]=-s[i];
            scanf("%d",&m);
            For(i,1,m)
            {
                int p,q;
                scanf("%d%d",&p,&q);
                edge[p].push_back(make_pair(q,i));
                edge[q].push_back(make_pair(p,-i));
            }
            if(sum)
            {
                printf("Impossible\n");
                continue;
            }
            printf("Possible\n");
            dfs(1,0,0);
            For(i,1,m)
            printf("%lld\n",ans[i]);
        }
    return 0;
}


你可能感兴趣的:(思维,dfs)