POJ 3277

POJ 3277

狂汗的一道题。本来这算是线段树入门级题目了，但是以下是我这个菜鸟的做题情况：

看了下用时最短的只有32MS，相形之下，我的前三次可以算做是TLE了。后来网上找了个代码改了改，时间才缩短到172MS。

我看了下我们的线段树构造主要有以下两个区别：

1．  我的线段树只存了每段的高度。其左边界和有边界在调用函数时写出来，即为以下形式：

insert(p,l,r,a,b){

   if(l==a&&r==b){...return;}

   m=(l+r)/2;

   if(b<=m)insert(2*p,l,m,a,b);

   else if(a>=m)insert(2*p+1,m,r,a,b);

   else{

      insert(2*p,l,m,a,m);

      insert(2*p+1,m,r,m,b);

   }

}

而它的则采用结构体的形式，初始建树就将线段树左右边界及中间点保存起来，以后做insert操作时不必再计算。

2．  我的线段树结点的高即为其真实的高。若左子树与右子树高不等，记其高为-1。而他的线段树的高并非其真实的高，当父节点的高大于子树的高时，子树的高即为父节点的高；当子树的高大于父节点时子树的高即为自身的高。也就是说一条从根到叶结点的路径中最大的高即为叶子结点的高。故统计时必须计算到叶结点。

以下是我先后提交的代码：

#include <cstdio>#include <iostream>#include <cstdlib>using namespace std;int A[40001],B[40001],H[40001],N=0,i,j=0,n=0,low,high,mid,LS[80002]={0},S[80001]={0},tree[300000]={0};int cmp(const void*a,const void*b){ return *((int *)a)-*((int *)b);}void insert(int p,int l,int r,int a,int b){ int m=(l+r)>>1; if(tree[p]<0){ if(b<=S[m])insert(2*p,l,m,a,b); else if(a>=S[m])insert(2*p+1,m,r,a,b); else{ insert(2*p,l,m,a,S[m]); insert(2*p+1,m,r,S[m],b); } } else if(H[i]>tree[p]){ if(S[l]==a&&S[r]==b){tree[p]=H[i];return;} tree[2*p]=tree[p]; tree[2*p+1]=tree[p]; if(b<=S[m])insert(2*p,l,m,a,b); else if(a>=S[m])insert(2*p+1,m,r,a,b); else{ insert(2*p,l,m,a,S[m]); insert(2*p+1,m,r,S[m],b); } tree[p]=-1; }}long long Count(int p,int l,int r){ if(tree[p]>=0)return (S[r]-S[l])*(long long)tree[p]; return Count(2*p,l,(l+r)>>1)+Count(2*p+1,(l+r)>>1,r);}int main(){ scanf("%d",&n); for(i=1;i<=n;i++){ scanf("%d%d%d",&A[i],&B[i],&H[i]); LS[++j]=A[i]; LS[++j]=B[i]; } qsort(LS+1,j,sizeof(int),cmp); for(i=1;i<=j;i++){ S[++N]=LS[i]; while(LS[i+1]==LS[i])i++; } for(i=1;i<=n;i++) insert(1,1,N,A[i],B[i]); cout<<Count(1,1,N)<<endl; return 0;}

#include <cstdio>#include <iostream>#include <algorithm>using namespace std;int A[40001],B[40001],H[40001],N=0,i,j=0,n=0,low,high,mid,LS[80002]={0},S[80001]={0};struct{ int l,r,h,m;}tree[300000]={0};void build(int p,int l,int r){ tree[p].l=l;tree[p].r=r; if(r-l>1){ int m=(l+r)>>1; tree[p].m=m; build(2*p,l,m); build(2*p+1,m,r); }}void insert(int p,int a,int b){ if(S[tree[p].l]==a&&S[tree[p].r]==b){ if(tree[p].h<H[i])tree[p].h=H[i]; return; } int tmp=S[tree[p].m]; if(b<=tmp)insert(2*p,a,b); else if(a>=tmp)insert(2*p+1,a,b); else{ insert(2*p,a,tmp); insert(2*p+1,tmp,b); }}long long Count(int p,int h){ if(h>tree[p].h)tree[p].h=h; if(tree[p].r-tree[p].l==1)return (long long)(S[tree[p].r]-S[tree[p].l])*tree[p].h; return Count(2*p,tree[p].h)+Count(2*p+1,tree[p].h);}int main(){ scanf("%d",&n); for(i=1;i<=n;i++){ scanf("%d%d%d",&A[i],&B[i],&H[i]); LS[++j]=A[i]; LS[++j]=B[i]; } sort(LS+1,LS+1+j); for(i=1;i<=j;i++){ S[++N]=LS[i]; while(LS[i+1]==LS[i])i++; } build(1,1,N); for(i=1;i<=n;i++) insert(1,A[i],B[i]); cout<<Count(1,0)<<endl; return 0;}

不知道到底是哪种原因造成了如此惊人的差距，我也懒得测试了。总之学会了线段树的另一种构造方法，算是不错的收获吧。