P6185 [NOI Online #1 提高组] 序列 题解

P6185 [NOI Online #1 提高组] 序列

题面：

题目背景

由于本题数据较难构造，所以无法保证卡掉所有错误做法。

题目描述

小 D 有一个长度为 $n$ 的整数序列 $a_{1\dots n}$，她想通过若干次操作把它变成序列 $b_i$。

小 D 有 $m$ 种可选的操作，第 $i$ 种操作可使用三元组 $(t_i,u_i,v_i)$ 描述：若 $t_i=1$，则她可以使 $a_{u_i}$ 与 $a_{v_i}$ 都加一或都减一；若 $t_i=2$，则她可以使 $a_{u_i}$ 减一、$a_{v_i}$ 加一，或是 $a_{u_i}$ 加一、$a_{v_i}$ 减一，因此当 $u_i=v_i$ 时，这种操作相当于没有操作。

小 D 可以以任意顺序执行操作，且每种操作都可进行无限次。现在给定序列与所有操作，请你帮她判断是否存在一种方案能将 $a_i$ 变为 $b_i$。题目保证两个序列长度都为 $n$。若方案存在请输出 YES，否则输出 NO。

输入格式

本题输入文件包含多组数据。

第一行一个正整数 $T$ 表示数据组数。对于每组数据：

第一行两个整数 $n,m$，表示序列长度与操作种数。

第二行 $n$ 个整数表示序列 $a_i$。

第三行 $n$ 个整数表示序列 $b_i$。

接下来 $m$ 行每行三个整数 $t_i,u_i,v_i$，第 $i$ 行描述操作 $i$。

注意：同一个三元组 $(t_i,u_i,v_i)$ 可能在输入中出现多次。

输出格式

对于每组数据输出一行一个字符串 YES 或 NO 表示答案。

样例 #1

样例输入 #1

3
1 1
1
3
1 1 1
2 3
1 2
4 5
1 1 2
2 1 2
1 1 2
3 3
1 2 3
5 5 4
1 1 2
1 1 3
2 2 3

样例输出 #1

YES
YES
YES

提示

样例 1 解释

第一组数据：使用一次操作 $1$。
第二组数据：使用三次操作 $1$。
第三组数据：使用三次操作 $1$，令 $a_1,a_2$ 都增加 $3$，再使用一次操作 $2$，令 $a_1,a_3$ 都增加 $1$。

---

数据范围与提示

对于测试点 $1\sim 5$：$n=2$，$m=1$，$a_i,b_i\le 99$，$u_1\ne v_1$，$t_1=1$。
对于测试点 $6\sim 10$：$n=2$，$m=1$，$a_i,b_i\le 99$，$u_1\ne v_1$，$t_1=2$。
对于测试点 $11\sim 12$：$n=2$，$a_i,b_i\le 99$，$u_i\ne v_i$。
对于测试点 $13\sim 16$：$t_i=2$。
对于测试点 $17$：$n,m\le 20$。
对于测试点 $18$：$n,m\le 10^3$。
对于所有测试点：$1\le T\le 10$，$1\le n,m\le 10^5$，$1\le a_i,b_i\le 10^9$， t i ∈ { 1 , 2 } t_i \in \{1,2\} ti​∈{1,2}，$1\le u_i,v_i\le n$。

二分图染色！NOIP2023考了

我们发现每个 ${\mathrm{opt}}_2$ 就像流水，可以在两个点中任意流转

那么，我们先将每个 ${\mathrm{opt}}_2$ 配合并查集连边缩点，

再记 $c_S={\sum }_{i\in S}{b}_i-a_i$，相同的本质就是让每一个 $c_S=0$

然后我们考虑对 ${\mathrm{opt}}_1$ 连边，形成一个图：

1. 若该图是二分图，两边同加同减，只要二分图两边总和相同即可

2. 若该图不是二分图，必然有一边内部有一条 ${\mathrm{opt}}_1$ 的边
   つまり，有一边或两边总和都可以进行 $+2$、$-2$ 的操作，
   而这种操作不会改变两边的奇偶性！故判断两边总和奇偶性是否相同！

AC-code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
int rd() {
	int x = 0, w = 1;
	char ch = 0;
	while (ch < '0' || ch > '9') {
		if (ch == '-') w = -1;
		ch = getchar();
	}
	while (ch >= '0' && ch <= '9') {
		x = x * 10 + (ch - '0');
		ch = getchar();
	}
	return x * w;
}

void wt(int x) {
	static int sta[35];
	int f = 1;
	if(x < 0) f = -1,x *= f;
	int top = 0;
	do {
		sta[top++] = x % 10, x /= 10;
	} while (x);
	if(f == -1) putchar('-');
	while (top) putchar(sta[--top] + 48);
}
constexpr int N = 2e5+5;
int n,m,a[N],b[N],d[N],s[N],co[N],p[N],top;
array<int,3> q[N];
int find(int x) {
	if(s[x] ^ x) s[x] = find(s[x]);
	return s[x];
}

int head[N],nxt[N << 1],to[N<<1],cnt;
void init() {
	for(int i = 0;i<=n;i++) head[i] = -1;
	for(int i = 0;i<=n;i++) d[i] = 0,s[i] = i,co[i] = 0;
	cnt = 0;top = 0;
}

void add(int u,int v) {
	nxt[cnt] = head[u];
	to[cnt] = v;
	head[u] = cnt++;
}

bool dfs(int x, int col) {
	co[x] = col, p[col] += d[x];
	bool ok = 1;
	for (int i = head[x];~i;i = nxt[i]) {
		int y = to[i];
		if (co[y] == co[x]) ok = 0;
		if (!co[y] && !dfs(y, 3 - col)) ok = 0;
	}
	return ok;
}


bool solve(){
	n = rd(),m = rd();
	for(int i = 1;i<=n;i++) a[i] = rd();
	for(int i = 1;i<=n;i++) b[i] = rd();
	init();
	for(int i = 1;i<=m;i++) {
		array<int,3> c;
		c[0] = rd(),c[1] = rd(),c[2] = rd();
		if(c[0] == 2) {
			int fx = find(c[1]),fy = find(c[2]);
			s[fx] = fy;
		}else q[++top] = c;
	}
	for(int i = 1;i<=n;i++) d[find(i)] += b[i] - a[i];
	for(int i = 1;i<=top;i++) {
		int fx = find(q[i][1]),fy = find(q[i][2]);
		add(fx,fy);add(fy,fx);
	}
	for(int i = 1;i<=n;i++) 
		if(find(i) == i && !co[i]) {
			p[1] = p[2] = 0;
			bool ok = dfs(i,1);
			if(ok && p[1] != p[2]) return false;
			if(!ok && ((p[1] ^ p[2]) & 1)) return false;
		}
	return true;

}

signed main() {
	int T = rd();
	while(T--) puts(solve() ? "YES" : "NO");
	return 0;
}