1. 程式人生 > >bzoj 3809 Gty的二逼妹子序列 —— 莫隊+分塊

bzoj 3809 Gty的二逼妹子序列 —— 莫隊+分塊

turn color ref sin 我想 com add get %d

題目:https://www.lydsy.com/JudgeOnline/problem.php?id=3809

據說一開始應該想到莫隊+樹狀數組,然而我想的卻是莫隊+權值線段樹...

如果用權值線段樹,則修改和查詢都是 O(logn),總復雜度 O(n√nlogn),艱難...(而且仔細一看空間有點卡?)

看了TJ,才發現權值也可以分塊,則查詢 O(√n) 但修改 O(1),就可以過咯~

代碼如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include
<cmath> using namespace std; int const xn=1e5+5,xm=1e6+5; int n,m,a[xn],blk[xn],c[xn],s[400],ans[xm]; struct N{int l,r,a,b,id;}q[xm]; bool cmp(N x,N y){return blk[x.l]==blk[y.l]?x.r<y.r:blk[x.l]<blk[y.l];} int rd() { int ret=0,f=1; char ch=getchar(); while(ch<0||ch>9){if(ch==-
)f=0; ch=getchar();} while(ch>=0&&ch<=9)ret=(ret<<3)+(ret<<1)+ch-0,ch=getchar(); return f?ret:-ret; } int query(int a,int b) { int ret=0; int x=blk[a],y=blk[b]; for(int i=x+1;i<y;i++)ret+=s[i]; if(x==y) { for(int i=a;i<=b;i++) if(c[i])ret++;
return ret; } for(int i=a;blk[i]==blk[a];i++) if(c[i])ret++; for(int i=b;blk[i]==blk[b];i--) if(c[i])ret++; return ret; } void add(int ps) { int x=a[ps]; if(!c[x])s[blk[x]]++; c[x]++; } void del(int ps) { int x=a[ps]; if(c[x]==1)s[blk[x]]--; c[x]--; } int main() { n=rd(); m=rd(); int bs=sqrt(n); for(int i=1;i<=n;i++)a[i]=rd(),blk[i]=(i-1)/bs+1; for(int i=1;i<=m;i++)q[i].l=rd(),q[i].r=rd(),q[i].a=rd(),q[i].b=rd(),q[i].id=i; sort(q+1,q+m+1,cmp); add(1); int l=1,r=1; for(int i=1;i<=m;i++) { int ql=q[i].l,qr=q[i].r; while(l<ql)del(l),l++; while(l>ql)l--,add(l); while(r<qr)r++,add(r); while(r>qr)del(r),r--; ans[q[i].id]=query(q[i].a,q[i].b); } for(int i=1;i<=m;i++)printf("%d\n",ans[i]); return 0; }

bzoj 3809 Gty的二逼妹子序列 —— 莫隊+分塊