【题面】
![在这里插入图片描述](https://i-blog.csdnimg.cn/direct/af04dda6529147a4bf218363edd7561a.png#pic_center)
【题目大意】
- 给定n和k,有一个长度为n的数列,对于区间
[
l
,
r
]
[l,r]
[l,r],当且仅当该区间内每个数字都在区间内出现k次时,该区间为好区间。
- 求该数列内一共有多少个好区间。
【解题思路】
- 对于数列中的每个位置都配置一个哈希值,使得每种数字的哈希值的周期为k,即从中任取连续k个相同的数字,这k个哈希值的异或和为0.
- 配置方法:对第0个位置生成随机数作为哈希值,然后对第1、2、3…个位置生成随机数并且异或前一个哈希值。对于第k-1个数,哈希值为第k-2个数和第0个数的异或和。
- 如此构造,即可使得任意k个哈希值的异或和都是0.
- 因此问题转化为:求有多少个区间,哈希值的异或和为0。本问题可以用前缀和+二分查找解决。
- 先求出所有哈希值的前缀和。当左、右端点前缀和之差为0时,意味着该区间每种数字都出现了k的倍数(可能出现多个周期)次。只要从左往右枚举右端点,用记录位置+二分查找 求出有多少个合法左端点就可以找到区间总数了。
- 考虑k的倍数的问题。只要在枚举右端点时,统计该位置数字出现的次数,当次数大于k,就意味着肯定不会成立,让左端点一直右移即可。
【参考程序】
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
#define int LL
int T,n,k,ans,
a[200005];
ULL sum[200005];
map < int , int > mp;
map < int , vector < ULL > > h;
map < int, vector < int > > slot;
signed main()
{
random_device rd;
unsigned seed = rd();
mt19937_64 rad(seed);
cin>>T;
while (T--)
{
scanf("%lld%lld",&n,&k);
ans=0;
h.clear();
slot.clear();
mp.clear();
slot[0].push_back(0);
for (int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
if (!mp.count(a[i]))
{
ULL x=rad(),y=rad();
h[a[i]].push_back(x);
h[a[i]].push_back(y);
sum[i]=sum[i-1]^x^y;
}
else
if (mp[a[i]]<k-1)
{
ULL x=rad();
sum[i]=sum[i-1]^x^h[a[i]].back();
h[a[i]].push_back(x);
}
else
{
int p=mp[a[i]]%k;
ULL cur=h[a[i]][p]^h[a[i]][(p+1)%k];
sum[i]=sum[i-1]^cur;
}
++mp[a[i]];
slot[sum[i]].push_back(i);
}
mp.clear();
for (int l=1,r=1;r<=n;r++)
{
mp[a[r]]++;
while (mp[a[r]]>k)
{
mp[a[l]]--;l++;
}
if (k==1)
{
ans+=r-l+1;continue;
}
auto v=slot[sum[r]];
int ll=lower_bound(v.begin(),v.end(),l-1)-v.begin(),
rr=lower_bound(v.begin(),v.end(),r)-v.begin();
ans+=rr-ll;
}
printf("%lld\n",ans);
}
return 0;
}