bzoj3262: 陌上花開 樹套樹
阿新 • • 發佈:2019-02-10
++i else while ins pair treap 註意 zoj block
鏈接
https://www.lydsy.com/JudgeOnline/problem.php?id=3262
思路
CDQ版本稍後再說
二維偏序排序用樹狀數組求就可以
而三維之後就不可以了,但我們可以在BIT上維護一個功能強大的treap
維護多出來的一維c
BIT套treap
a排序
BIT維護b
treap維護c
錯誤
需要註意的是,重復元素就不能挨個求了,他們的貢獻是最後一個的貢獻*數量,註意思考
代碼
#include <iostream> #include <algorithm> #include <ctime> #include <cstdlib> #include <cstdio> using namespace std; const int N=2e5+7; int read() { int x=0,f=1;char s=getchar(); for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1; for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0'; return x*f; } #define AA first #define BB second.first #define CC second.second pair<int,pair<int,int> > x[N]; int n,m; namespace treap { int ch[N*40][2],val[N*40],pri[N*40],siz[N*40],cnt; int make_new(int k) { val[++cnt]=k,pri[cnt]=rand(),siz[cnt]=1; return cnt; } void pushup(int rt) { siz[rt]=siz[ch[rt][0]]+siz[ch[rt][1]]+1; } int merge(int x,int y) { if(!x||!y) return x+y; if(pri[x]<pri[y]) { ch[x][1]=merge(ch[x][1],y); pushup(x); return x; } else { ch[y][0]=merge(x,ch[y][0]); pushup(y); return y; } } void split(int now,int &x,int &y,int k) { if(!now) x=y=0; else { if(val[now]<=k) x=now,split(ch[now][1],ch[x][1],y,k); else y=now,split(ch[now][0],x,ch[y][0],k); pushup(now); } } int k_th(int now,int k) { while(233) { if(siz[ch[now][0]]+1==k) return now; if(siz[ch[now][0]]>=k) now=ch[now][0]; else k-=siz[ch[now][0]]+1,now=ch[now][1]; } } void insert(int &now,int w) { int x,y; split(now,x,y,w); now=merge(merge(x,make_new(w)),y); } int query(int now,int k) { int x,y; split(now,x,y,k); int ans=siz[x]; now=merge(x,y); return ans; } } namespace BIT { int rt[N]; void add(int b,int c) { for(int i=b;i<=m;i+=(i&-i)) treap::insert(rt[i],c); } int query(int b,int c) { int ans=0; for(int i=b;i>=1;i-=(i&-i)) ans+=treap::query(rt[i],c); return ans; } } int ans[N]; int main() { n=read(),m=read(); for(int i=1;i<=n;++i) x[i].AA=read(),x[i].BB=read(),x[i].CC=read(); sort(x+1,x+1+n); int js=1; for(int i=1;i<=n;++i) { if(x[i]==x[i+1]) js++; else ans[BIT::query(x[i].BB,x[i].CC)]+=js,js=1; BIT::add(x[i].BB,x[i].CC); } for(int i=0;i<n;++i) printf("%d\n",ans[i]); return 0; }
bzoj3262: 陌上花開 樹套樹