hdu_2795_线段树好题

//最主要的一点是:为什么会想到用线段树(非常关键)???牛人说的,哈!太神了,用一般方法超时了,我试过!其实跟huffman的解码过程很像//此题不像一般的线段树一样给出query区间[a,b],只要遍历到node[i].l>=a&&node[i].r<=b即可,而是要遍历到叶子节点//对h进行离散化,每一个区间的key值保存这个区间的每个叶子所剩w的最大值,query和update合为一步,其他的没什么好说的//time:1960ms#include<iostream>#include<cstdio>#include<cmath>using namespace std;struct tree{int l,r,w;}node[600005];int h,w,n;int max(int a,int b){return a>b?a:b;}int min(int a,int b){return a>b?b:a;}void creat(int i,int a,int b){node[i].l=a;node[i].r=b;node[i].w=w;if(a==b)return ;else{int mid=(a+b)/2;creat(2*i,a,mid);creat(2*i+1,mid+1,b);}}int query_update(int i,int key) //深搜要一直到达叶子{int temp;if(node[i].w>=key){if(node[i].l==node[i].r){node[i].w-=key;return node[i].l;}else if(node[2*i].w>=key){temp=query_update(2*i,key);}else{temp=query_update(2*i+1,key);}node[i].w=max(node[2*i].w,node[2*i+1].w); //保存左右孩子中最大的那个return temp;}return -1;}int main(){while(scanf("%d%d%d",&h,&w,&n)!=EOF){int m=min(h,200000); //如果超过20w就用20w,因为每一个小广告的高是1，最多也就是20w的高度int wx;int index;creat(1,1,m);while(n--){scanf("%d",&wx);printf("%d/n",query_update(1,wx));}}return 0;}