CF1129D Isolation

题面传送门
蛮好的一道dp题。
显然设\(dp_i\)为到\(i\)的方案数。设\(f(l,r)\)为\([l,r]\)区间内出现一次的个数，那么有方程\(dp_i=\sum\limits_{f(j+1,i)\leq k}{dp_j}\)
这样是\(O(n^2)\)的看上去没法优化。
考虑数颜色的套路就是维护前面最近的。设其为\(pre_i\)，那么当\(pre_i\)加入时，\([pre_{pre_i},pre_i-1]\)范围内的要\(-1\),\([pre_i,i-1]\)范围内要\(+1\)
那么这个问题就变成了区间\(val\)加减\(1\)，查询\(val_i\leq k\)的\(dp_i\)值。
这个用线段树什么的很难维护，但是用分块开个桶记录即可。
时空复杂度\(O(n\sqrt n)\)。但是这道题卡空间，所以要稍微开大一点块长。
code:

#include<cstdio>
#include<cmath>
#define mod 998244353
#define N 100039
#define ll long long
#define I inline
#define re register
using namespace std;
int lazy[N],k,ks,n,a[N],b[N],g[N],pre[N];ll dp[N],f[129][N*2],tot;
I void add(int x){re int now=x/ks;tot+=dp[x];f[now][N-lazy[now]]+=dp[x];b[x]=-lazy[now];}
I void gets(int l,int r){
	re int now=l/ks,i;
	for(i=l;i<=r;i++) tot-=(b[i]+lazy[now]==k?dp[i]:0),f[now][b[i]+N]-=dp[i],b[i]++,f[now][b[i]+N]+=dp[i];
}
I void get(int l,int r){
	if(l/ks==r/ks)return gets(l,r);gets(l,l/ks*ks+ks-1);gets(r/ks*ks,r);
	for(re int i=l/ks+1;i<r/ks;i++) tot-=f[i][k-lazy[i]+N],lazy[i]++;
}
I void dels(int l,int r){
	re int now=l/ks,i;
	for(i=l;i<=r;i++) f[now][b[i]+N]-=dp[i],b[i]--,tot+=(b[i]+lazy[now]==k?dp[i]:0),f[now][b[i]+N]+=dp[i];
}
I void del(int l,int r){
	if(l/ks==r/ks)return dels(l,r);dels(l,l/ks*ks+ks-1);dels(r/ks*ks,r);
	for(re int i=l/ks+1;i<r/ks;i++) lazy[i]--,tot+=f[i][k-lazy[i]+N];
}
int main(){
	freopen("1.in","r",stdin);
	re int i;scanf("%d%d",&n,&k);ks=sqrt(n*9);
	for(i=1;i<=n;i++) scanf("%d",&a[i]);
	for(i=1;i<=n;i++) pre[i]=g[a[i]],g[a[i]]=i;dp[0]=1;add(0);
	for(i=1;i<=n;i++)pre[i]&&(del(pre[pre[i]],pre[i]-1),0),get(pre[i],i-1),tot%=mod,dp[i]=tot,add(i);
	printf("%lld\n",(dp[n]+mod)%mod);
}