【CF241E】Flights(差分约束)

【CF241E】Flights(差分约束)

题面

CF
\(n\)个点\(m\)条边,要求给每条边赋一个\(1\)\(2\)的边权,判断能否使得每一条\(1\)\(n\)的路径的权值和都相等,如果可以给出一个方案。

题解

首先有这样一个结论:从\(1\)号点到达任意一个点的所有路径的权值都相同。比较显然,就不证明了。
然后就可以直接差分约束了?

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
using namespace std;
#define MAX 1010
inline int read()
{
    int x=0;bool t=false;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=true,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return t?-x:x;
}
struct Line{int v,next,w;}e[MAX*10];
int h[MAX],cnt=1;
inline void Add(int u,int v,int w){e[cnt]=(Line){v,h[u],w};h[u]=cnt++;}
vector<int> E1[MAX],E2[MAX];
int n,m,U[MAX*5],V[MAX*5],vis[MAX],dis[MAX],num[MAX];bool inq[MAX];
queue<int> Q;
void dfs1(int u){vis[u]|=1;for(int v:E1[u])if(!(vis[v]&1))dfs1(v);}
void dfs2(int u){vis[u]|=2;for(int v:E2[u])if(!(vis[v]&2))dfs2(v);}
int main()
{
    n=read();m=read();
    for(int i=1;i<=m;++i)
    {
        int u=read(),v=read();U[i]=u;V[i]=v;
        E1[u].push_back(v);E2[v].push_back(u);
    }
    dfs1(1);dfs2(n);
    for(int i=1;i<=m;++i)
        if(vis[U[i]]==3&&vis[V[i]]==3)
        {
            Add(U[i],V[i],1);
            Add(V[i],U[i],-2);
        }
    memset(dis,-63,sizeof(dis));
    Q.push(1);inq[1]=true;dis[1]=0;
    while(!Q.empty())
    {
        int u=Q.front();Q.pop();
        for(int i=h[u];i;i=e[i].next)
        {
            int v=e[i].v;
            if(dis[v]<dis[u]+e[i].w)
            {
                dis[v]=dis[u]+e[i].w;
                if(!inq[v])inq[v]=true,Q.push(v),++num[v];
            }
        }
        if(num[u]>=n){puts("No");return 0;}
        inq[u]=false;
    }
    puts("Yes");
    for(int i=1;i<=m;++i)
        if(vis[U[i]]==3&&vis[V[i]]==3)printf("%d\n",dis[V[i]]-dis[U[i]]);
        else puts("1");
    return 0;
}
posted @ 2019-07-08 17:08 小蒟蒻yyb 阅读(...) 评论(...) 编辑 收藏