Two Hundred Twenty One (hard version)(思维/前缀和/二分)

题目
题意:符串的贡献为正,偶数位上的字符串贡献为逆(+符号做-贡献,-符号做+贡献),字符串的权值为字符串上所有位的贡献和。有若干次查询,每次查询区间[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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值