bzoj4500: 矩陣（dfs）

題目傳送門
。

解法：
d[i]表示這一行/列需要變動多少。
那麼如果我問到一個已經有d[i]的且它的改動值不能滿足限制的話就輸出NO就行了呀。

其實一個點的值就等於行的改動值+列的改動值。
dfs即可。

程式碼實現：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
struct node {int x,y,c,next;}a[5100];int len,last[2100];
void ins(int x,int y,int c) {len++;a[len].x=x;a[len].y=y;a[len].c=c;a[len].next=last[x];last[x]=len;}
bool v[2100];int d[2100];
bool dfs(int x) {
    v[x]=true;
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(v[y]==false) {d[y]=a[k].c-d[x];if(dfs(y)==false)return false;}
        else if(d[y]!=a[k].c-d[x]) return false;
    }return true;
}
int main() {
    int T;scanf("%d",&T);
    while(T--) {
        int n,m,k;scanf("%d%d%d",&n,&m,&k);len=0;memset(last,0,sizeof(last));
        for(int i=1;i<=k;i++) {
            int x,y,c;scanf("%d%d%d",&x,&y,&c);ins(x,y+n,c);ins(y+n,x,c);
        }bool bk=true;memset(v,false,sizeof(v));
        for(int i=1;i<=n+m;i++) if(v[i]==false) if(dfs(i)==false) {bk=false;break;}
        if(bk==false) printf("No\n");else printf("Yes\n");
    }
    return 0;
}