D. Prefix Permutation Sums-优快云博客

本文链接：https://blog.youkuaiyun.com/hacker_51/article/details/132790615

文章讨论了如何通过计算排列1到n的前n项和，并利用去重操作确定给定一组前缀和是否能唯一确定原数组的问题。通过比较剩余元素数量和理论上的可能性，判断是否能找到唯一解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

样例：

输入
`12 5 6 8 12 15 5 1 6 8 15 4 1 2 100 4 1 3 6 2 2 3 1 2 4 3 7 10 5 5 44 46 50 4 1 9 10 5 13 21 36 42 5 1 2 3 1000000000000000000 9 9 11 12 20 25 28 30 33`

输出
YES YES NO YES YES NO YES NO NO NO NO NO

思路：

数学问题，根据前缀和与排列1 ~ n 之间的关系来推导。

代码详解如下：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
#include <set>
#include <unordered_map>
#define endl '\n'
#define int long long
#define YES puts("YES")
#define NO puts("NO")
#define umap unordered_map
#pragma GCC optimize(3,"Ofast","inline")
#define ___G std::ios::sync_with_stdio(false),cin.tie(0), cout.tie(0)
using namespace std;
const int N = 2e6 + 10;

inline void solve()
{
	int n;
	cin >> n;
	
	int s = (1 + n)*n / 2;	// 排列 1 ~ n 的前n项和
	
	// 去重操作，由于我们的排列是1 ~ n 不会有多余的元素
	set<int>b;
	
	int x,r = 0;	// 这里的 x 是输入的当前前缀和元素，r 是记录上一个元素
	for(int i = 1;i < n;++i)
	{
		cin >> x;
		
		// 当前的前缀和减去上一个的前缀和，
		// 得到可能的 原来 数组 元素a 
		// 存储好所得的结果
		if(x - r <= n)
		{
			b.insert(x - r);
		}
		
		r = x;	// 更新上一个元素
	}
	
	// 由于我们是知道 n - 1 个前缀和元素，
	// 所以当我们还原 a 元素，就会得到 n - 2 个可能的 a 元素
	// 所以 b.size() == n - 2 又因为倘若我们最后的前缀和元素 x != s
	// 那么排列一定有可能推出来，因为前面肯定会丢失，
	// 当我们丢失的，只要是有一个它们的相隔超过 n 于是得到的是 (n - 1) - 2 
	// 然后当我们 x != s 的时候 n - 1 + 1 - 2 == n - 2
	if(b.size() == n - 2 + (x != s)) YES;
	else NO;
}


signed main()
{
//	freopen("a.txt", "r", stdin);
	___G;
	int _t = 1;
	cin >> _t;
	while (_t--)
	{
		solve();
	}

	return 0;
}