Codeforces Round #741 (Div. 2)D2 - Two Hundred Twenty One (hard version)

题目大意：
给你t组数据
每组数据给你一个n与q
代表有n个长度只包含’+‘或’-'的字符串序列
+代表1
-代表-1
区间和定义为

q个询问给你l,r，让你删除[l,r]区间的一些数，使得更新后的区间和为0

思路：
二分
首先我们要知道区间和等于0的时候我们要删几个数（这就是D1做的事情)
我们设d=[l,r]区间和
如果d==0的时候那肯定直接输出0就行了
而如果d是奇数的时候
我们用个前缀和数组ci表示[l,i]区间和
那么cr=d
假设现在i位置的符号是’+’
那么|ci-ci-1|=1
我们现在假设删掉一个位置i
即ci-1-(cr-ci)(注意c表示前缀和,删掉位置后面那段会取反)
然后我们就找
ci-1-(cr-ci)=0
如果这个时候设这里符号×(-1)ⁱ值为ai=1
即2×ci-1=cr=d的情况
那么我们就找到了该删去符号的位置i
并且i满足ai=1,ci=(d+1)/2
同时注意(d+1)/2的值是一定会出现的
实际上[1,d]的整数值都会出现
因为前面说了|ci-ci-1|=1
那么1要么一直+1加到d，要么中途会有几个-1之后再+1到d
但因为它是一步一步+1到d的
那么[1,d]的整数值肯定一步一步都会一一出现
如果ai是-1也是类似做法
我们利用三个下标一个存ai的值,一个存前缀和的值,一个存i
然后我们在对应的ai,ci二分找大于等于l的值即可
如果d是偶数输出r后转为求[l,r-1]奇数即可
同时要说明一下
当区间长度为奇数的时候d为奇数
偶数的时候d为偶数
首先长度相同的时候改变一个符号的时候都是进行+2,-2的操作
奇偶性不会变化
而长度为1的时候为1或-1
长度为2的时候为2或-2或0
那么就说明区间长度奇偶性与d奇偶性一致了
具体见代码

AC代码

#include <bits/stdc++.h>

using namespace std;
const int N=3e5+10;
int sum[N];
char s[N];
vector<int>v[2][N*2];
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,q;
        scanf("%d%d",&n,&q);
        for(int i=0;i<=n*2+1;i++){
            v[1][i].clear();
            v[0][i].clear();
        }
        sum[0]=n+1;//防止负数情况
        scanf("%s",s+1);
        for(int i=1; i<=n; i++)
        {
            sum[i]=sum[i-1]+((i&1)?1:-1)*(s[i]=='+'?1:-1);
            if(i&1)v[s[i]=='+'?1:0][sum[i]].push_back(i);//记得符号乘上-1
            else v[s[i]=='+'?0:1][sum[i]].push_back(i);
        }
        while(q--)
        {
            int l,r;
            int d;
            scanf("%d%d",&l,&r);
            if(sum[r]==sum[l-1]){
                    printf("0\n");
                    continue;
            }
            else if((r-l+1)&1)
            {
                printf("1\n");
                d=sum[r]-sum[l-1];
            }
            else
            {
                printf("2\n");
                printf("%d ",r);
                d=sum[r-1]-sum[l-1];
            }
            int sum1=(d+1)/2+sum[l-1];//注意要加上sum[l-1]
            if(d<0)
            {
                d=-d;
                sum1=sum[l-1]-(d+1)/2;
                printf("%d",v[0][sum1][lower_bound(v[0][sum1].begin(),v[0][sum1].end(),l)-v[0][sum1].begin()]);//二分找大于等于l的位置
            }
            else printf("%d",v[1][sum1][lower_bound(v[1][sum1].begin(),v[1][sum1].end(),l)-v[1][sum1].begin()]);
            puts("");
        }
    }
    return 0;
}