海报贴的优先度:
① 高 > 低 || 高相等 && 左 > 右
朴素的想法:两层for循环,每贴一张海报考察从第1行到第 min(h,n) 行有没有空余的位置能够贴的下,若贴的下,- = w ;这样的复杂度为n^2,(2*1e5)^2,超时!
学习了线段树之后,很自然的想到,将考察操作用线段树来完成,这样就缩减到了nlogn;
线段树中每个节点保存其左右孩子的可用宽度的最大值,初始都为w;
观察二叉树的性质,如果 seg[i].value>w,即树中是存在满足贴海报的条件的,要按照 条件① 贴海报,当然是不停的先递归其左子树,左子树不满足再递归右子树。
这里的updata操作可以合并到query操作里面(因为query操作每次都是查询到最底层,updata也是从最底层开始),即在回溯的时候更新父亲节点的max值。
代码如下
<span style="font-size:12px;">#include <iostream> #include <stdio.h> #include <algorithm> using namespace std; #define maxn 200005 int a[maxn]; int h,w,n; struct node { int left,right; int value; }seg[4*maxn]; void build(int i,int l,int r) { seg[i].value=w; seg[i].left=l; seg[i].right=r; if(seg[i].left==seg[i].right) return ; int mid=(l+r)>>1; build(i<<1,l,mid); build(i<<1|1,mid+1,r); } int query(int i,int x) { if(seg[i].left==seg[i].right) { seg[i].value-=x; return seg[i].left; } int ret; if(seg[i<<1].value >= x) ret=query(i<<1,x); else ret=query(i<<1|1,x); seg[i].value=max(seg[i<<1].value,seg[i<<1|1].value); return ret; } int main() { int x; while(~scanf("%d%d%d",&h,&w,&n)) { build(1,1,min(n,h)); for(int i=1;i<=n;i++) { scanf("%d",&x); if(seg[1].value < x) printf("-1\n"); else printf("%d\n",query(1,x)); } } return 0; } </span>