带权并查集 then

问题 B: 便
时间限制: 2 Sec 内存限制: 512 MB
提交: 50 解决: 14
题目描述
给出一个R*C的棋盘.共有R行C列,R*C个格子.现要在每个格子都填一个非负整数.使得任意一个2*2的正方形区域都满足这样的性质:左上角的数字+右下角的数字=左下角的数字+右上角的数字.有些格子已经确定,你不能更改其中的数字.其他格子的数字由你决定.
不难验证每个2*2的区域都是符合要求的.
Orbitingflea想要知道一个可行的填充棋盘的方案.但是这个方案可能很大.所以你只需对给定的棋盘判定是否存在至少一种可行的填充棋盘的方案.
输入
第一行输入一个T，表示数据组数。接下来T组数据。
每组数据的第1行2个整数R,C表示棋盘的大小.
第2行1个整数n表示已经被填好数字的格子的数目.
接下来n行每行3个整数ri,ci,ai,表示第ri行ci列的格子被填上了数字ai.
输出
T行.第i行是第i组数据的答案.有合法方案时输出一行Yes,没有时输出一行No.

样例输入
6
2 2 3
1 1 0
1 2 10
2 1 20

2 3 5
1 1 0
1 2 10
1 3 20
2 1 30
2 3 40

2 2 3
1 1 20
1 2 10
2 1 0

3 3 4
1 1 0
1 3 10
3 1 10
3 3 20

2 2 4
1 1 0
1 2 10
2 1 30
2 2 20

1 1 1
1 1 -1
样例输出
Yes
No
No
Yes
No
No
提示
第1个测试点,R=1
第2,3个测试点,R*C<=12,如果有解,保证存在一个解使得所有数字大小不超过2
第4,5个测试点,R=2
第6,7个测试点,R=3
第8个测试点,1<=R,C<=20
第9个测试点,1<=R,C<=100
对于全部测试点,1<=T<=6,1<=R,C,n<=100000,1<=ri<=R,1<=ci<=C,同一个格子不会多次被填上数字.ai是整数且绝对值不超过10^9.

这道题翻译成中文竟然叫 便。。。。→_→
对于这个棋盘，要满足l1+r2==l2+r1,换言之l1-l2==r1-r2.从左向右以此类推。
因此，对于某一行，它与另一行对应列的差值相等。那么，这就和并查集很像了。可以通过已确定的点，实现行列间的的比较。而且差具有传递性，所以完全可以用带权并查集。
那么举个例子，行1和行2确定了差值，行2和行3确定了，就可以间接确定行1和行3的差，就把行3连成行1的儿子。
如果有合并，此处考虑两个问题。
1，如果当前两行已经有同一个父亲，那么如果他们的当前差不等于他们与父亲确定的差值的差，那么就可以返回无解了。
2，对于父亲的合并cha[fx]=cha[x,y]+cha[y]-cha[x];其实也挺好理解的。。
这个处理完后，那么就不会再存在互相矛盾的情况了，但还有一种情况，就是出现值小于零。那么如何判断呢？
对于所有确定值的点，那么它可以给父亲行提供一个值，（因为行与行之间的差确定了，那么确定的点值就可以给父亲行对应列一个确定的值），那么求出这个最小值。然后找出父亲与儿子最大的那个差。这样就找到了此联通块中的最小值了。判断是否小于零即可。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define N 100000
#define ll long long 
using namespace std;
int read()
{
    int sum=0,f=1;char x=getchar();
    while(x<'0'||x>'9'){if(x=='-')f=-1;x=getchar();}
    while(x>='0'&&x<='9')sum=sum*10+x-'0',x=getchar();
    return sum*f;
}
int r,c,n,p=1,t;
int fx[N+5],fy[N+5];
ll wx[N+5],wy[N+5],hh1[N+5],hh2[N+5];
struct node
{
    int x,y; ll h;
    node(){}inline node(int x_,int y_,ll h_){x=x_;y=y_;h=h_;}
}a[N+5];
inline bool cmp1(node a,node b){return a.x<b.x;}
inline bool cmp2(node a,node b){return a.y<b.y;}
inline int find_x(int x)
{
    if(fx[x]==x)return fx[x];
    int y=find_x(fx[x]);
    wx[x]+=wx[fx[x]];
    return fx[x]=y;
}
inline int find_y(int x)
{
    if(fy[x]==x)return fy[x];
    int y=find_y(fy[x]);
    wy[x]+=wy[fy[x]];
    return fy[x]=y;
}
inline int hb_x(int x,int y,ll l)
{
    int a=find_x(x),b=find_x(y);
    if(a==b)return wx[x]==l+wx[y];
    fx[a]=b;
    wx[a]=l+wx[y]-wx[x];
    return 1;
}
inline int hb_y(int x,int y,ll l)
{
    int a=find_y(x),b=find_y(y);
    if(a==b)return wy[x]==l+wy[y];
    fy[a]=b;
    wy[a]=l+wy[y]-wy[x];
    return 1;
}
inline int work()
{
    for(int i=0;i<=N+2;i++)fy[i]=fx[i]=i;
    //for(int i=0;i<=c;i++)fy[i]=i;
    memset(wx,0,sizeof(wx));
    memset(wy,0,sizeof(wy));
    memset(hh1,120,sizeof(hh1));
    memset(hh2,120,sizeof(hh2));
    r=read();c=read();n=read();p=1;
    int x,y,z;
    for(int i=1;i<=n;i++)
    {
        x=read();y=read();z=read();
        a[i]=node(x,y,z);
        if(z<0||(r*c<=12&&z>=3&&r!=1))p=0;//
    }
    if(p==0)return 0;
    sort(a+1,a+n+1,cmp1);
    for(int i=1;i<n;i++)
       if(a[i].x==a[i+1].x&&!hb_y(a[i].y,a[i+1].y,a[i+1].h-a[i].h))
           return 0;
    sort(a+1,a+n+1,cmp2);
    for(int i=1;i<n;i++)
       if(a[i].y==a[i+1].y&&!hb_x(a[i].x,a[i+1].x,a[i+1].h-a[i].h))
           return 0;
    for(int i=1;i<=n;i++)
    {
        int x=find_x(a[i].x);
        hh1[x]=min(hh1[x],a[i].h+wx[a[i].x]);
    }
    for(int i=1;i<=r;i++)
    {
        int x=find_x(i);
        hh2[x]=min(hh2[x],-wx[i]);
    }
    for(int i=1;i<=r;i++)
        if(fx[i]==i&&hh1[i]+hh2[i]<0)
            return 0;
    return 1;
}
int main()
{
    t=read();
    while(t--)
    {
        if(!work())printf("No
");
        else printf("Yes
");
    }
}