C2. Make Nonzero Sum (hard version) codeforces1754C2

Problem - C2 - Codeforces

题目大意：有一个长度为n只由0,1,-1组成的数组，要求把这个数组分割，使得每一部分按照a[l]-a[l+1]+a[l+2]..的规则运算后对每一块求和，要求和是0，求一种合法的分割方式

1<=n<=2e5

思路：我们从最小的块开始想，如果这个块的大小是1，那么他对答案的贡献就是本身，如果是2，那么贡献就是al-ar，然后因为我们要将所有块求和，所以如果我们当前有两块，分别为1,-1，那么直接分成两块即可，然后如果是1,1这样两个数相同的，就把他们分成一块，贡献就是0，对于0直接自己成块即可，同时由上述结论可知一个1或-1必须需要另一个1或-1才能变成0，所以如果1和-1的总数为奇数，肯定是没有合法分割的，对于其他情况，我们统计1和-1的位置和值，对于每两个值，如果不等，就分别成一块，如果相等且相邻就分为一块，如果中间有0的话，就让右边的值和左边的一个0成一块即可，这样就使得l和r不同了，对于多余的0直接使相邻的0成一块即可

#include<iostream>
#include<cstdio>
using namespace std;
const int N = 2e5 + 5;
pair<int, int> ans[N];
int a[N];
pair<int, int>b[N];
int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		int n;
		int cnt = 0;
		scanf("%d", &n);
		int last, now;
		int cntb = 0;//统计1和-1的数量
		for (int i = 1; i <= n; i++)
		{
			scanf("%d", &a[i]);
			if (a[i] != 0)
			{
				b[++cntb] = make_pair(a[i], i);//记录1和-1的值，位置
			}			
		}
		if (cntb & 1)
		{//1和-1的总数不能是奇数
			printf("-1\n");
			continue;
		}
		if (cntb>=1&&b[1].second >= 2)
		{//先把起始的0存入答案
			ans[++cnt] = make_pair(1, b[1].second - 1);
		}
		for (int i = 1; i <= cntb; i += 2)
		{//遍历每两个1或0
			last = b[i].first;
			now = b[i + 1].first;
			if (last == now)
			{//两个值如果相等
				if (b[i].second + 1 != b[i + 1].second)
				{//两个数原始位置不相邻
					ans[++cnt] = make_pair(b[i].second, b[i].second);//l自己一块
					if (b[i + 1].second >= b[i].second + 3);
					{
						ans[++cnt] = make_pair(b[i].second + 1, b[i + 1].second - 2);
					}//中间的0一块
					ans[++cnt] = make_pair(b[i + 1].second - 1, b[i + 1].second);//r和r-1一块
				}
				else
				{
					ans[++cnt] = make_pair(b[i].second, b[i + 1].second);//相邻的直接两个数成一块
				}
			}
			else
			{//两个值不等
				ans[++cnt] = make_pair(b[i].second, b[i].second);//l自成一块
				if (b[i + 1].second >= b[i].second + 2)
				{
					ans[++cnt] = make_pair(b[i].second + 1, b[i + 1].second - 1);
				}//中间的0一块
				ans[++cnt] = make_pair(b[i + 1].second, b[i + 1].second);//r自成一块
			}
			if (i + 2 <= cntb && b[i + 2].second >= b[i + 1].second + 2)
			{//两组数中间的0成一块
				ans[++cnt] = make_pair(b[i + 1].second + 1, b[i + 2].second - 1);
			}
		}
		if (n >= b[cntb].second + 1)
		{
			ans[++cnt] = make_pair(b[cntb].second + 1, n);
		}//末尾的0成一块
		printf("%d\n", cnt);
		for (int i = 1; i <= cnt; i++)
		{
			printf("%d %d\n", ans[i].first, ans[i].second);
		}
	}
	return 0;
}