原文链接https://www.cnblogs.com/zhouzhendong/p/AtCoder-SoundHound-Inc-Programming-Contest-2018-E.html
题目传送门 - AtCoder SoundHound Inc. Programming Contest 2018 E
题意
给定一个无向连通图,有 $n$ 个节点 $m$ 条带权边,第 $i$ 条边连接 $x_i,y_i$ ,权值为 $s_i$ ,没有重边、自环。
现在,请你给每一个节点取一个正整数点权。问有多少种方案使得任意一条边两端的节点权值和等于边权。
$2leq nleq 100000,1leq mleq 100000$
所有输入的数字都在 $10^9$ 以内。
题解
先吐槽:
这题细节好坑啊!!!我当场做到只 WA 一个点,没想到 20 分钟还是没有发现特判,然后 GG 。然后考完发现在我没注意的地方忘记特判了??然后考完不到10分钟把它过了。就加了几行。
然后讲做法。
设 $v_i$ 为第 $i$ 个点的点权。
首先,我们考虑到对于所有的 $i$ ,有 $v_{x_i}+v_{y_i}=s_i$ 。我们把式子移动一下,得到:
$$v_{x_i}-s_i=(-v_{y_i})$$
$$(-v_{y_i})+s_i=v_{x_i}$$
$$v_{y_i}-s_i=(-v_{x_i})$$
$$(-v_{x_i})+s_i=v_{y_i}$$
我们使节点 $1$ 作为初始节点,即令 $v_1=alpha$ 。
我们考虑将每一个点拆成两个点,一个点记录其正的权值(即 $v_i=alpha + k$ 时,记录的值为 $k$ ),另一个点记录其负权值(即 $-v_i=alpha+k$ ,记录的值为 $k$)。
然后我们对于每一条边,拆成上述四条有向边。
然后 bfs 一遍把每一个点与 $alpha$ 的关系求出来。这里注意一点,如果到达一个点有两条距离不同的路径,那么显然答案为 $0$ 。(条件冲突)
然后我们得到了一些数据。
我们考虑去解决那些拆点之后两个节点都被访问的节点。
对于每一个这样的节点,我们可以解出唯一的 $alpha$ ,如果所有节点的解有不同,那么答案显然是 $0$ 。否则答案显然是 $1$ 。
请您先思索一下在选中下面黑色矩形区域内的字看下面的话。
这样是错的!!我就是挂在这里了。我们不能这么着急的确定答案是 $1$ 。因为我们还需要满足所有点权均为正整数。所以我们还需要判一判。
然后就只剩下二分图的情况了。
对于这种情况,我们只需要根据每一个节点与初始节点 $1$ 的关系,根据“正整数”这个条件更新 $alpha$ 的取值范围。最后输出即可。
代码
#include <bits/stdc++.h> using namespace std; typedef long long LL; const int N=400005; struct Gragh{ int cnt,y[N],z[N],nxt[N],fst[N]; void clear(){ cnt=0; memset(fst,0,sizeof fst); } void add(int a,int b,int c){ y[++cnt]=b,z[cnt]=c,nxt[cnt]=fst[a],fst[a]=cnt; } }g; int n,m; int q[N],head,tail; LL dis[N]; LL INF=10000000000000000LL; void out0(){ puts("0"); exit(0); } void SPFA(int S){ for (int i=1;i<=n*2;i++) dis[i]=INF; head=tail=0; q[++tail]=S; dis[S]=0; while (head!=tail){ int x=q[++head],y; for (int i=g.fst[x];i;i=g.nxt[i]){ int y=g.y[i]; if (dis[y]!=dis[x]+g.z[i]){ if (dis[y]!=INF) out0(); dis[y]=dis[x]+g.z[i]; q[++tail]=y; } } } } int main(){ scanf("%d%d",&n,&m); g.clear(); for (int i=1;i<=m;i++){ int a,b,c; scanf("%d%d%d",&a,&b,&c); g.add(a,b+n,-c); g.add(b+n,a,c); g.add(b,a+n,-c); g.add(a+n,b,c); } SPFA(1); LL v=INF; for (int i=1;i<=n;i++) if (dis[i]!=INF&&dis[i+n]!=INF){ LL A=dis[i],B=dis[i+n]; if ((A+B)%2LL) out0(); LL x=-(A+B)/2LL; if (x!=v) if (v==INF) v=x; else out0(); } if (v!=INF){ int f=1; for (int i=1;i<=n;i++){ if (dis[i]!=INF) if (v+dis[i]<=0) f=0; if (dis[i+n]!=INF) if (v+dis[i+n]>=0) f=0; } printf("%d",f); return 0; } LL MIN=1,MAX=INF; for (int i=1;i<=n;i++) if (dis[i]!=INF) MIN=max(MIN,-dis[i]+1); else MAX=min(MAX,-dis[i+n]-1); printf("%lld",max(MAX-MIN+1,0LL)); return 0; }