模板题:
三维偏序。
一维CDQ一维离线排序解决一维树状数组。
细节问题。这个题的偏序可以相等就很恶心。
注意离线排序和CDQ还有树状数组都不能完美的解决有相等元素互相贡献的问题,只会由小标号贡献到大标号(视实现方式而定),这个可以事后用一个
m
a
p
map
map把相等的贡献加回去。但是不相等,因为是一层层剥开的偏序,所以在外层相等时,一定要让内层小的放外面。
A C C o d e AC\ Code AC Code
#pragma GCC optimize(3)
#include<bits/stdc++.h>
#define maxn 100005
using namespace std;
int n,K;
int x[maxn],y[maxn],z[maxn],c[maxn],a[maxn],cnt[maxn],ans[maxn],tr[maxn*2],c2[maxn];
map<tuple<int,int,int>,int>mp;
inline bool cmp(const int &u,const int &v){
return z[u] == z[v] ? x[u] == x[v] ? y[u] == y[v] ? u<v : y[u] < y[v] : x[u] < x[v] : z[u] < z[v];
}
inline void upd(int u,int v){ for(;u<=K;u+=u&-u) tr[u]+=v; }
inline int qsum(int u){ int ret=0;for(;u;u-=u&-u) ret+=tr[u];return ret; }
void Solve(int l,int r){
if(l>=r) return;
int mid = (l+r) >> 1;
Solve(l,mid),Solve(mid+1,r);
for(int i=l,j=mid+1,k=l;(i<=mid || j<=r);){
if(j>r || (i<=mid && x[c[i]]<=x[c[j]])){
upd(y[c[i]],1);
a[k++] = c[i++];
}
else{
ans[c[j]]+=qsum(y[c[j]]);
a[k++] = c[j++];
}
}
for(int i=l;i<=mid;i++) upd(y[c[i]],-1);
for(int i=l;i<=r;i++) c[i] = a[i];
}
int main(){
scanf("%d%d",&n,&K);
for(int i=1;i<=n;i++)
scanf("%d%d%d",&x[i],&y[i],&z[i]),c[i]=i;
for(int i=n;i>=1;i--)
ans[i] += mp[make_tuple(x[i],y[i],z[i])]++;
sort(c+1,c+1+n,cmp);
Solve(1,n);
for(int i=1;i<=n;i++) cnt[ans[i]]++;
for(int i=0;i<n;i++) printf("%d\n",cnt[i]);
}
UPD:
树状数组套树状数组解法:
A
C
C
o
d
e
\mathcal AC \ Code
AC Code
#include<bits/stdc++.h>
#define maxn 200005
#define rep(i,j,k) for(int i=(j),LIM=(k);i<=LIM;i++)
#define pb push_back
#define mp make_pair
using namespace std;
int n,K;
struct node{ int a,b,c; bool operator <(const node &B)const{ return a == B.a ? b == B.b ? c < B.c : b < B.b : a < B.a; }}A[maxn];
int tr[maxn],vis[maxn],ans[maxn],f[maxn],g[maxn],tim;
void upd(int u,int v){ for(;u<maxn;u+=u&-u) (vis[u] != tim) && (vis[u] = tim , tr[u] = 0) , tr[u] += v; }
int qry(int u){ int r=0;for(;u;u-=u&-u) if(vis[u] == tim) r += tr[u]; return r; }
vector<pair<int,int> >Q[maxn];
int main(){
scanf("%d%d",&n,&K);
rep(i,1,n) scanf("%d%d%d",&A[i].a,&A[i].b,&A[i].c);
sort(A+1,A+1+n);
rep(i,1,n){
for(int u=A[i].b;u;u-=u&-u) Q[u].pb(mp(i,0));
for(int u=A[i].b;u<maxn;u+=u&-u) Q[u].pb(mp(A[i].c,1));
}
for(tim=1;tim<maxn;tim++) for(auto u:Q[tim])
if(u.second == 0) f[u.first] += qry(A[u.first].c);
else upd(u.first,1);
for(int i=n;i>=1;i--) g[i] = (A[i].a == A[i+1].a && A[i].b == A[i+1].b && A[i].c == A[i+1].c) ? g[i+1] + 1 : 1 , f[i] += g[i] - 1;
rep(i,1,n) ans[f[i]] ++;
rep(i,0,n-1) printf("%d\n",ans[i]);
}