题目
Easy_Version思路
很容易发现,我们将点 i i i去掉;
对于 [ L , i − 1 ] [L,i-1] [L,i−1]是没有影响的,而对于 [ i + 1 , R ] [i+1,R] [i+1,R]是乘了个 − 1 -1 −1;
我们设 f ( i ) f(i) f(i)表示将点 i i i抹去后的区间和是多少;
f ( i ) = s ( i − 1 ) − s ( l − 1 ) − s ( r ) + s ( i ) , s 是 前 缀 和 数 组 f(i)=s(i-1)-s(l-1)-s(r)+s(i),s是前缀和数组 f(i)=s(i−1)−s(l−1)−s(r)+s(i),s是前缀和数组
我们想让这个区间为0,我们令 f ( i ) = 0 f(i)=0 f(i)=0
可以得到 s ( r ) − s ( i ) = s ( i − 1 ) − s ( l − 1 ) s(r)-s(i)=s(i-1)-s(l-1) s(r)−s(i)=s(i−1)−s(l−1)
也就是图中的左右区间相等;
我们可以想到,这个区间和必然需要是一个奇数;
因为只有奇数,才能将中间某个点去掉,将两边平分;
又因为我们只移除了一个点,因此当区间和为奇数时,ans=1
;
如果区间和是偶数,那么我们可以删除最后一个位置上的元素,现在又转化成了区间和为奇数的情况;
因此ans=2
;
Code
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 3e5 + 10;
char str[N];
int s[N];
void solve(){
int n,q;
cin >> n >> q;
cin >> (str+1);
for(int i=1;i<=n;++i){
if(str[i] == '+') s[i] = 1;
else s[i] = -1;
if(i%2 == 0) s[i]*=-1;
s[i] += s[i-1];
}
int L,R,sum;
while(q--){
cin >> L >> R;
sum = s[R] - s[L-1];
if(sum == 0) cout << 0 << '\n';
else if(sum&1) cout << 1 << '\n';
else cout << 2 << '\n';
}
}
int main(){
std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
int t;
cin >> t;
while(t--){
solve();
}
return 0;
}
Hard_Version思路
上回书说到…我们已经知道要找到一个点 i i i使得下式子成立;
s ( r ) − s ( i ) = s ( i − 1 ) − s ( l − 1 ) s(r)-s(i)=s(i-1)-s(l-1) s(r)−s(i)=s(i−1)−s(l−1)
即 s ( r ) + s ( l − 1 ) = s ( i ) + s ( i − 1 ) s(r)+s(l-1)=s(i)+s(i-1) s(r)+s(l−1)=s(i)+s(i−1)
因为等号左边我们是可以 O ( 1 ) O(1) O(1)的时间求出来的,但是等号右边没那么好搞;
我们可以考虑对于每个 s ( i ) + s ( i − 1 ) s(i)+s(i-1) s(i)+s(i−1),用一个 s e t set set来维护;
每次二分查找这个点 i i i即可;
Code
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <unordered_map>
#include <set>
using namespace std;
typedef long long ll;
const int N = 3e5 + 10;
char str[N];
int s[N],a[N];
unordered_map<int,set<int>> um;
int find_in_set(int L,int R){
int tot = s[L-1] + s[R];
//要抹去点,范围必须在[L,R]中;
//[L,R]里面必有答案,因此找大于等于L的即可
return *um[tot].lower_bound(L);
}
void work(){
int L,R,sum;
cin >> L >> R;
sum = s[R] - s[L-1];
if(sum == 0){
cout << 0 << '\n';
return;
}
if(sum&1){
cout << 1 << '\n';
cout << find_in_set(L,R) << '\n';
return;
}else{
cout << 2 << '\n';
cout << R <<' '<<find_in_set(L,R-1) << '\n';
return;
}
}
void solve(){
um.clear();
int n,q;
cin >> n >> q;
cin >> (str+1);
for(int i=1;i<=n;++i){
if(str[i] == '+') s[i] = 1;
else s[i] = -1;
if(i%2 == 0) s[i]*=-1;
s[i] += s[i-1];
}
for(int i=1;i<=n;++i){
int tot = s[i] + s[i-1];
um[tot].insert(i);
}
while(q--){
work();
}
}
int main(){
std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
int t;
cin >> t;
while(t--){
solve();
}
return 0;
}