【Ybtoj】【 NOIP2021 模拟赛 B 组 Day4】B. 循环数组

---

解题思路

对于一个数x，只要一个区间中出现了x，而不论它的个数是多少，它对这个区间权值的贡献都只是1。

那么我们考虑枚举每一个x来求出含有它的区间的个数，然而直接求很麻烦，我们运用补集转化的思想，求出不含x的区间个数，用总区间个数减去这一部分的值，即为x的贡献。

我们假设得到了x在这个n×k的数组中出现的位置分别为$p1,p2,\cdots ,pm$

因为对于一个长度为x的序列，它总共有$x(x+1)/2$的子区间（一个数也算做一个区间），所以我们设$f(x)=x(x+1)/2$，那么可以得到不含x的区间总共有$f(p1-1)+f(n\times k-pm)+{\sum }_i^{m-1}=1f(pi+1-pi-1)$。

首先$f(p_1-1)$和$f(n\times k-p_m)$都是可以直接O(1)计算出来的，唯一比较麻烦的就是${\sum }_{i=1}^{m-1}f(p_{i+1}-p_{i-1})$这一部分。为了方便描述，我们就把这个$n\times k$的长度的数组描述为k个长度为n的数组。

我们通过观察是可以发现这$个p_{i+1}-p_i$只可能有两种情况，第一种是$p_{i+1}$和$p_i$在同一个长度为n的数组中，这种情况总共会被计算k次（在每一个数组中都会被计算一次），另一种情况就是pi是一个长度为n的数组中x这个数最后出现的位置，而$p_i+1$则是x在下一个长度为n的数组中第一次出现的位置，而这种情况则会被计算$k-1$次。

---

代码

#include<cstdio>
#include<iostream>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<map>
#define ll long long
using namespace std;

const int mod=1e9+7;
int n,k,a[100010];
ll ans;

map<int,int>p,v;

ll calm(ll x,ll y)
{
	ll w=y-x+1; 
	return w*(w+1)/2%mod;
}

int main(){
	freopen("loop.in","r",stdin);
	freopen("loop.out","w",stdout);
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		if(!p.count(a[i]))
			ans=(ans+calm(1LL,1LL*n*k%mod))%mod,ans=(ans+mod-calm(1,i-1))%mod;
		else ans=(ans+mod-calm(p[a[i]]+1,i-1)*k%mod)%mod;
		p[a[i]]=i;
	}
	for(int i=1;i<=n;i++)
	{
		if(!v.count(a[i]))
		{
			ans=(ans+mod-calm(p[a[i]]+1,n)-calm(p[a[i]]+1,n+i-1)*(k-1)%mod)%mod;
			v[a[i]]=1;
		}
	}
	printf("%d",ans); 
}