ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

CF1129D Isolation

2021-04-10 21:33:50  阅读:180  来源: 互联网

标签:pre lazy int CF1129D Isolation now dp define


题面传送门
蛮好的一道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);
}

标签:pre,lazy,int,CF1129D,Isolation,now,dp,define
来源: https://www.cnblogs.com/275307894a/p/14642175.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有