?
题外话:本次考试基本都是数据结构题目。博主感觉做起来很顺......当然WA是不可避免的......
题目大意:一个字符串只包含+ 和 -两种字符,满足要求的子串或者是+和-数量相同,如+-,++--。或者是-号与+号的数量差为3的倍数,因为两个-可以升级成promising一个+。
这样问题就是求某个子串,其中-号数量减去+号数量是3的倍数。
解题思路:我们把+号看成-1,减号看成1。这样问题就变成求一个序列中有多少个子段,其“子段和大于等于0且为3的倍数”。
本题目还有个F1 (easy)版本,长度小于3000,可以双循环暴力枚举子段[i,j],用前缀和预处理[i,j]这一段的结果,判断是否满足条件,复杂度O(n^2)。
本题解法:如果一个子段右端点是i,那么如何找到满足条件的左端点,根据前缀和的思想距离,例如sum[i]=8,那么只要左端点前结果是8,5,2-1,-4.....,sum[i]减去这些值结果一定是3的倍数,这些值的特点是对3求余均为2。插播一个小知识,负数求余直接%结果是错的哦,如果解决请参看下方代码。因此当处理到第i个字符时,如果此时sum[i]=X,那么i左侧有多少个位置,其前缀和<=X,同时与X对3同余,这些位置都能和i组成满足条件的字段。
这样为题就转换成求sum[i]左边有多少个比它小且同余的sum[?]数量。由于递推过程中,不同数组的数量也在不断变化,这种可变的区间和问题可用树状数组或线段树解决。至于如何处理同余,因为对3求余结果只有3种可能,我这里开了3个树状数组,分别存储余数为0,1,2的对应数据。根据余数到对应树状数组求和。
#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
int t,n,sum[3][800005];
char s[200005];
int lowbit(int x)/**< lowbit的作用请草稿纸模拟 */
{
return x&(-x);
}
void update(int pos,int P) /**< 树状数组模板 */
{
for(; pos<=n+n; pos+=lowbit(pos))
sum[P][pos]++;
}
int getsum(int pos,int P)
{
int s=0;
for(;pos>0;pos-=lowbit(pos))
s+=sum[P][pos];
return s;
}
int main()
{
ios::sync_with_stdio(0),cin.tie(0);
int i,j;
cin>>t;
while(t--)
{
int cnt=0;ll ans=0;
cin>>n>>s+1;
for(i=0;i<=4*n;i++)/**< 树状数组初始化 */
sum[0][i]=sum[1][i]=sum[2][i]=0;
update(0+n,0);/**< 这个处理区间[1...i]满足条件的情况 */
for(i=1; i<=n; i++)
{
cnt=cnt+(s[i]=='+'?-1:1);
ans+=getsum(cnt+n+1,(cnt%3+3)%3);/**< 找到比cnt小的数字个数 */
update(cnt+n+1,(cnt%3+3)%3);/**< 把数据存到树状数组,因为可能是负数,所以加一个偏移量 */
/**< 切记不能出现0,不然死循环,我开始写的cnt+n,连挂几次 */
}
cout<<ans<<endl;
}
return 0;
}
|