题目
题意:符串的贡献为正,偶数位上的字符串贡献为逆(+符号做-贡献,-符号做+贡献),字符串的权值为字符串上所有位的贡献和。有若干次查询,每次查询区间[l,r]的字符串,需要删掉最少多少字符,才能使该子串的权值刚好为0。要求把要删除的位置下标输出。
官方题解
思路:同easy version。现在多出了要输出对应的下标,由于b数组是有序的,我们可以直接二分查找下标。
#include<bits/stdc++.h>
using namespace std;
const int maxn = 300010;
typedef long long ll;
char s[maxn];
int n, q, l, r;
int a[maxn];
int pre[maxn];
void init() {
pre[0] = 0;
for (int i = 1; i <= n; ++i) {
a[i] = s[i] == '+' ? 1 : -1;
if ((i & 1) && s[i] == '+' ||
!(i & 1) && s[i] == '-') {
pre[i] = pre[i-1] + 1;
} else {
pre[i] = pre[i-1] - 1;
}
}
}
int get_sum(int l, int r) {//获取区间和
if (l > r) return 0;
int sum = pre[r] - pre[l-1];
if (l % 2 == 0) sum = -sum;
return sum;
}
int check(int l, int r, int m) {// 计算删了下标m后的区间[l,r]的和
if (!((l & 1) ^ (m & 1))) {// 判断l和m下标的奇偶性是否相同
return get_sum(l, m - 1) + get_sum(m + 1, r);
}
return get_sum(l, m - 1) - get_sum(m + 1, r);
}
int find1(int L, int R) {
int l = L, r = R;
while (l < r) {
int m = (l + r) / 2;
int left = check(L, R, l);
int right = check(L, R, r);
int res = check(L, R, m);
if(!left) return l;
if (!right) return r;
if (!res) return m;
if (!((left < 0) ^ (res < 0))) {// 判断left和res是否同号
l = m;
} else {
r = m;
}
}
return l;
}
int main() {
int t;
scanf("%d", &t);
while (t--) {
scanf("%d%d", &n, &q);
scanf("%s", s + 1);
init();
while (q--) {
scanf("%d%d", &l, &r);
int res = -1;
vector<int> ve;
if (!(pre[r] - pre[l-1])) {
res = 0;
} else if ((r - l + 1) & 1) {
res = 1;
ve.push_back(find1(l, r));
} else {
res = 2;
ve.push_back(l);
ve.push_back(find1(l + 1, r));
}
printf("%d\n", res);
for (auto u: ve) {
printf("%d ", u);
}
if (ve.size()) printf("\n");
}
}
return 0;
}