Codeforces Global Round 15 D. Array Differentiation（按位枚举 + 思维）

链接 D. Array Differentiation

题意

给出一个大小为 n 的 a 序列，问是否有一个 b 序列满足 a[i] = b[j] - b[k]；

思路

因为 a[i] = b[j] - b[k] 且 a 序列和 b 序列的大小一样，先将 a 序列中的一个数用 b₁ 和 b₂ 去构造，然后剩余的数我们用新的 b_i 和 b₁ b₂ 中的一个数去构造，当 b 序列构造完 n 个后，发现 a 序列中还有一个数不知满足不满足题中的条件，所以可知如果满足条件，那么剩余的那个 a 序列的数一定是由 b 中的某两个已经确定的数构造的；

这样把 b 序列看作点权，a 序列看作边权，可以知道当构成一个环的时候 b 序列是满足条件的；

假如说 a₁ = b₁ - b₂，a₂ = b₂ - b₃，那么，如果a₃ = b₃ - b₁，那么我们可以发现，序列 a 中的三个数，我们只用了 b 序列中的三个数就构造完成了，进一步观察发现 a₁ + a₂ + a₃ = 0；
所以可以得出一个结论，当 t₁a₁ + t₂a₂ +…t_ma_m = 0（t为符号）时，满足条件；

（b₁ - b₂）+（b₂ - b₃）+⋯+（b_m - b₁）= 0；

给 a 序列中的元素分配符号，加上，减去和不取，当它们的和为0的时候可以肯定是可以构造成功的；

注意，最后一种情况是不用枚举的，因为最后一种情况是每一个数都不取，那是不合法的；

AC代码

#include <bits/stdc++.h>
#define hz020 return
#define mes memset
#define mec memcpy
#define x first
#define y second
#define pb push_back
#define all(x) (x).begin(), (x).end()

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll,ll> PII;

const double pi = acos(-1);
const int N = 20;
const int null = 0x3f3f3f3f,INF = 1e9;
const ll mod = 998244353;

int T;
int n;
int a[N];

int main()
{
	cin >> T;
	while (T --)
	{
		cin >> n;

		int res = 1;
		for (int i = 1; i <= n; i ++) cin >> a[i], res *= 3;
		int flag = 0;
		for (int i = 1; i <= res - 1; i ++)
		{
			int t = i, sum = 0;

			for (int i = 1; i <= n; i ++)
			{
				if (t % 3 == 1) sum += a[i];
				if (t % 3 == 2) sum -= a[i];

				t /= 3;
			}
			if (!sum)
			{
				flag = 1;
				break;
			}
		}
		if (flag) puts("YES");
		else puts("NO");
	}
	
	hz020 0;	
}

END