【三维偏序】【分块】bzoj3262 陌上花开

裸的三维偏序。 对x坐标排序，y、z坐标分块。复杂度O(n*sqrt(n*log(n)))。代码很短。

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
struct Point{int x,y,z,num;void Read(){scanf("%d%d%d",&x,&y,&z);}}p[100002];
bool operator < (const Point &a,const Point &b){return a.x<b.x;}
bool cmp (const Point &a,const Point &b){return a.y<b.y;}
vector<int>b[400];
vector<Point>a[400];
int n,rank[100001],m,head,maxv[400],sum;
void makeblock()
{
    int sz=(int)sqrt((double)n*(log((double)n)/log(2.0))); if(!sz) sz=1;
    for(sum=1;sum*sz<n;sum++)
      {
        int R=sum*sz;
        for(int i=(sum-1)*sz+1;i<=R;i++) p[i].num=sum;
        maxv[sum]=p[R].y;
      }
    for(int i=(sum-1)*sz+1;i<=n;i++) p[i].num=sum;
    maxv[sum]=p[n].y;
}
void insert(const Point &U)
{
    b[U.num].insert(upper_bound(b[U.num].begin(),b[U.num].end(),U.z),U.z);
    a[U.num].push_back(U);
}
int query(const Point &U)
{
    int cnt=0,i;
    for(i=1;i<=sum && maxv[i]<=U.y;++i)
      cnt+=upper_bound(b[i].begin(),b[i].end(),U.z)-b[i].begin();
    for(vector<Point>::iterator it=a[i].begin();it!=a[i].end();++it)
      if((*it).z<=U.z&&(*it).y<=U.y) ++cnt;
    return cnt;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) p[i].Read();
    sort(p+1,p+n+1,cmp); makeblock();
    sort(p+1,p+n+1);
    for(int i=1;i<=n;i++)
      {
          if(p[i].x!=p[i-1].x) head=i;
        if(p[i].x!=p[i+1].x)
          {
              for(int j=head;j<=i;j++) insert(p[j]);
              for(int j=head;j<=i;j++) ++rank[query(p[j])-1];
          }
      }
    for(int i=0;i<n;i++) printf("%d\n",rank[i]);
    return 0;
}