pku 3277 (线段树+离散化)

因为区间最大到10的9次方，开这么大的空间内存肯定不够，所以要离散化，用map存入然后用iterator遍历得到的有序序列存入vector。

然后以vector的下标建立线段树，统计时若结点不是叶子结点，则它的值为左右孩子的值之和，否则返回 底*高 。

 

参考 岳云涛 《浅谈线段树在信息学竞赛中的应用》

 

#include <iostream> #include <map> #include <vector> using namespace std; typedef struct nod { int left,right,mid; long long h; bool cover; }node,*nd; typedef struct p { long long start,end,hi; }point; // 元素个数上限和存储数组 const int MAX_NUM=200000; //第一次交re,干脆开个大的 node seg_tree[3*MAX_NUM+1]; point list[40005]; map<long long , int> my_map; map<long long , int>::iterator it; vector<long long> vec; // 创建 , l为左端点，r为右端点，num为在数组中的编号 void make(int l,int r,int num) { seg_tree[num].left=l; seg_tree[num].right=r; seg_tree[num].h=0; seg_tree[num].mid= (l+r)/2; if( l+1 != r ) { make(l,seg_tree[num].mid,2*num); make(seg_tree[num].mid,r,2*num+1); } } // 插入操作，参数意义同上 void Insert(int l,int r,int num ,long long &hi) { if( seg_tree[num].left==l && seg_tree[num].right==r ) { if(seg_tree[num].h<hi) seg_tree[num].h=hi; return ; } if( r<=seg_tree[num].mid ) Insert(l,r,2*num,hi); else if( l>=seg_tree[num].mid ) Insert(l,r,2*num+1,hi); else { Insert(l,seg_tree[num].mid,2*num,hi); Insert(seg_tree[num].mid,r,2*num+1,hi); } } //统计操作 long long cal(long long hi,int num) { if(hi>seg_tree[num].h) seg_tree[num].h=hi; if( seg_tree[num].left+1 == seg_tree[num].right ) { return seg_tree[num].h*( vec[ seg_tree[num].right ] - vec[ seg_tree[num].left ] ); } return cal( seg_tree[num].h , 2*num ) + cal( seg_tree[num].h , 2*num+1 ); } int main() { int T,t=1; cin>>T; long long a,b,h; for(int i=0;i<T;++i) { cin>>a>>b>>h; my_map[a]=0; my_map[b]=0; list[i].start=a; list[i].end=b; list[i].hi=h; } vec.push_back(0); //压入一个元素使vec从下标1开始有效 for( it=my_map.begin(); it!=my_map.end();++it ) { it->second=t++; vec.push_back( it->first ); } vec.push_back( vec[vec.size()-1] +1 ); //将离散化之后的元素按顺序压入向量 make(1,my_map.size()+1,1); for(int i=0;i<T;++i) Insert( my_map[ list[i].start ] , my_map[ list[i].end ] ,1 , list[i].hi ); cout<<cal( seg_tree[1].h ,1 )<<endl; return 0; }