BZOJ 3262 cdq分治 OR 树套树
注意判断 三个条件都一样的……
(CDQ分治 其实并不是很难理解 只是想不到……)
CDQ分治:
//By SiriusRen
#include <cstdio>
#include <algorithm>
using namespace std;
#define N 888888
int n,k,tree[N],tot,ans[N];
struct Node{int a,b,c,ans,sum;}node[N];
bool cmp(Node a,Node b){if(a.a!=b.a)return a.a<b.a;if(a.b!=b.b)return a.b<b.b;return a.c<b.c;}
bool cmp2(Node a,Node b){return a.b<b.b;}
void insert(int l,int r,int pos,int num,int wei){
if(l==r){tree[pos]+=wei;return;}
int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
if(mid>=num)insert(l,mid,lson,num,wei);
else insert(mid+1,r,rson,num,wei);
tree[pos]=tree[lson]+tree[rson];
}
int query(int l,int r,int pos,int num){
if(r<=num)return tree[pos];
int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
if(mid>=num)return query(l,mid,lson,num);
else return query(l,mid,lson,num)+query(mid+1,r,rson,num);
}
void cdq(int l,int r){
if(l==r)return;
int mid=(l+r)>>1,jy=l;
cdq(l,mid),cdq(mid+1,r);
sort(node+l,node+mid+1,cmp2),sort(node+mid+1,node+r+1,cmp2);
for(int i=mid+1;i<=r;i++){
while(jy<=mid&&node[jy].b<=node[i].b)insert(1,k,1,node[jy].c,node[jy].sum),jy++;
node[i].ans+=query(1,k,1,node[i].c);
}
while(jy-->l)insert(1,k,1,node[jy].c,-node[jy].sum);
}
int main(){
scanf("%d%d",&n,&k);
for(int i=1;i<=n;node[i].sum=1,i++)scanf("%d%d%d",&node[i].a,&node[i].b,&node[i].c);
sort(node+1,node+1+n,cmp);
for(int i=1;i<=n;i++)
if(node[i].a!=node[tot].a||node[i].b!=node[tot].b||node[i].c!=node[tot].c)node[++tot]=node[i];
else node[tot].sum++;
cdq(1,tot);
for(int i=1;i<=tot;i++)ans[node[i].ans+node[i].sum-1]+=node[i].sum;
for(int i=0;i<n;i++)printf("%d\n",ans[i]);
}
树套树 (线段树+treap):
//By SiriusRen
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 6005000
int n,k,root[N],size,t,ans,rec,Ans[1000500];
struct Node{int s,c,m;}node[1000500];
struct Tree{int ch[2],v,cnt,rnd,sz;}tr[N];
bool cmp(Node a,Node b){if(a.s!=b.s)return a.s<b.s;if(a.c!=b.c)return a.c<b.c;return a.m<b.m;}
void Upd(int k){tr[k].sz=tr[k].cnt+tr[tr[k].ch[0]].sz+tr[tr[k].ch[1]].sz;}
void rot(int &k,bool f){int t=tr[k].ch[f];tr[k].ch[f]=tr[t].ch[!f],tr[t].ch[!f]=k,Upd(k),Upd(t),k=t;}
void ins(int &k,int num){
if(!k){k=++size;tr[k].cnt=tr[k].sz=1,tr[k].rnd=rand(),tr[k].v=num;return;}
tr[k].sz++;
if(tr[k].v==num){tr[k].cnt++;return;}
bool f=num>tr[k].v;ins(tr[k].ch[f],num);
if(tr[k].rnd>tr[tr[k].ch[f]].rnd)rot(k,f);
}
void ask(int &k,int num){
if(!k)return;
if(tr[k].v==num){ans+=tr[k].cnt+tr[tr[k].ch[0]].sz; return;}
else if(tr[k].v>num)ask(tr[k].ch[0],num);
else ans+=tr[k].cnt+tr[tr[k].ch[0]].sz,ask(tr[k].ch[1],num);
}
void Insert(int l,int r,int pos){
ins(root[pos],node[t].m);
if(l==r)return;
int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
if(mid<node[t].c)Insert(mid+1,r,rson);
else Insert(l,mid,lson);
}
void Query(int l,int r,int pos){
if(r<=node[t].c){ask(root[pos],node[t].m);return;}
int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
if(mid>=node[t].c)Query(l,mid,lson);
else Query(l,mid,lson),Query(mid+1,r,rson);
}
int main(){
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
scanf("%d%d%d",&node[i].s,&node[i].c,&node[i].m);
sort(node+1,node+1+n,cmp);
for(t=1;t<=n;t++){
if(node[t].c!=node[t+1].c||node[t].m!=node[t+1].m||node[t].s!=node[t+1].s)
ans=0,Query(1,k,1),Ans[ans]+=rec+1,Insert(1,k,1),rec=0;
else rec++,Insert(1,k,1);
}
for(int i=0;i<n;i++)printf("%d\n",Ans[i]);
}