CodeForces - 1562D2 Two Hundred Twenty One (hard version)（ 前缀和 + 构造 + 二分 ）

题目链接：点击进入

题目

题意

长度为 n 的字符串，’+’ 表示 1 ，‘-’ 表示 -1 。q 次询问区间，每次问最少移除多少个字母以及被移除的位置。可以使区间内电荷为 0 。电荷的计算相当于对区间 [ l , r ] 排序，然后求 $a[1]-a[2]+a[3]-a[4]+...+(-1{)}^{r-l}\ast a[r-l+1]=0$ 。

思路

根据这个题的思路。
奇数长度，我们就是找一个中间的 pos ( pos >= l )，满足 sum [ r ] - sum [ pos ] = sum [ pos -1 ] - sum [ l - 1] ，转换一下，即 sum [ r ] + sum [ l - 1 ] = sum [ pos - 1 ] + sum [ pos ] 。因此，对于 sum [ i - 1 ] + sum [ i ] 的值，保存对应的位置 i ，查询 [ l , r ] 时，直接二分查找 sum [ r ] + sum [ l - 1 ] 值对应的所有位置中 >= l 的最小的那个位置即可。
偶数长度，第一个删除位置 l ,然后 l = l + 1 ，剩下的区间为奇数，按照上面处理即可。

代码

// Problem: D2. Two Hundred Twenty One (hard version)
// Contest: Codeforces - Codeforces Round #741 (Div. 2)
// URL: https://codeforces.com/contest/1562/problem/D2
// Memory Limit: 512 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<stack>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define base 233 
#define pb push_back
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
//#define int long long
//#define double long double
//#define rep(i,x,y) for(register int i = x; i <= y;++i)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=2e6+10;
const int mod=1e9+7;
const double eps=1e-9;
const int N=5e3+10;
/*--------------------------------------------*/
inline int read()
{
    int k = 0, f = 1 ;
    char c = getchar() ;
    while(!isdigit(c)){if(c == '-') f = -1 ;c = getchar() ;}
    while(isdigit(c)) k = (k << 1) + (k << 3) + c - 48 ,c = getchar() ;
    return k * f ;
}
/*--------------------------------------------*/

int t,n,q,a[maxn],sum[maxn];
char s[maxn];
vector<int>v[maxn];

int solve(int l,int r)
{
	int x=sum[r]+sum[l-1]+2*n;
	int pos=lower_bound(v[x].begin(),v[x].end(),l)-v[x].begin();
	return v[x][pos];
}

int main() 
{
	// ios::sync_with_stdio(false);
	// cin.tie(0);cout.tie(0);
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&q);
		scanf("%s",s);
		for(int i=1;i<=n;i++)
		{
			if(s[i-1]=='+')
				a[i]=1;
			else
				a[i]=-1;
		}
		for(int i=1;i<=n;i++)
		{
			if(i&1)
				sum[i]=a[i]*1;
			else
				sum[i]=a[i]*(-1);
			sum[i]+=sum[i-1];
		}
		for(int i=1;i<=n;i++)
			v[sum[i]+sum[i-1]+2*n].pb(i);
		while(q--)
		{
			int l,r;
			scanf("%d%d",&l,&r);
			if(sum[r]-sum[l-1]==0) cout<<0<<endl;
			else if((r-l+1)&1)
				printf("1\n%d\n",solve(l,r));
			else
				printf("2\n%d %d\n",l,solve(l+1,r));
		}
		for(int i=1;i<=n;i++)
			v[sum[i]+sum[i-1]+2*n].clear();
		for(int i=1;i<=n;i++) sum[i]=0;
	}
    return 0;
}