区间哈希——D-Interval Selection

【题面】

【题目大意】

• 给定n和k，有一个长度为n的数列，对于区间$[l,r]$，当且仅当该区间内每个数字都在区间内出现k次时，该区间为好区间。
• 求该数列内一共有多少个好区间。

【解题思路】

• 区间哈希算法。

1. 对于数列中的每个位置都配置一个哈希值，使得每种数字的哈希值的周期为k，即从中任取连续k个相同的数字，这k个哈希值的异或和为0.
   • 配置方法：对第0个位置生成随机数作为哈希值，然后对第1、2、3…个位置生成随机数并且异或前一个哈希值。对于第k-1个数，哈希值为第k-2个数和第0个数的异或和。
   • 如此构造，即可使得任意k个哈希值的异或和都是0.
2. 因此问题转化为：求有多少个区间，哈希值的异或和为0。本问题可以用前缀和+二分查找解决。
3. 先求出所有哈希值的前缀和。当左、右端点前缀和之差为0时，意味着该区间每种数字都出现了k的倍数（可能出现多个周期）次。只要从左往右枚举右端点，用记录位置+二分查找 求出有多少个合法左端点就可以找到区间总数了。
4. 考虑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;
}