http://poj.org/problem?id=2828
题意:有n个的排队,每一个人都有一个val来对应,每一个后来人都会插入当前队伍的某一个位置pos。要求把队伍最后的状态输出。
逆向思考。这样考虑,最后一个人一定会得到当前队伍他想要的位置,如果我们往前一个阶段,倒数第二个人也一定能得到他想要的位置……,也就是说,我们可以这样处理,我们把最后一个人插入,然后忽略它,再把倒数第二个人插入。即,我们找出当前队伍他想要插入的位置pos的真正坐标就可以。然后去更新整个队伍的长度。如此循环,直到最后一个人。线段树在单点更新的时候,感觉和二分查找是很相似的,可以用它实现。
#include <stdio.h>
#include <iostream>
#include <string>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <vector>
#define Mid(a,b) ( a+((b-a)>>1))
#define ll(x) (x<<1)
#define rr(x) (x<<1|1)
const int N = 200010;
using namespace std;
int n;
int pos[N];
int val[N];
int ans[N];
struct node
{
int left;
int right;
int sum;
int mid() { return Mid(left, right); }
};
struct segtree
{
node tree[N * 4];
void buildtree(int left,int right,int ind)
{
tree[ind].left = left;
tree[ind].right = right;
tree[ind].sum = right - left + 1;
if (left !=right)
{
int mid = tree[ind].mid();
buildtree(left,mid,ll(ind));
buildtree(mid+1,right,rr(ind));
}
}
int query(int val,int pos)
{
int left = tree[pos].left;
int right = tree[pos].right;
if (left == right)
{
tree[pos].sum = 0;
return left;
}
else
{
int ind;
if (tree[ll(pos)].sum >= val)
ind = query(val, ll(pos));
else
ind = query(val - tree[ll(pos)].sum , rr(pos));
tree[pos].sum = tree[ll(pos)].sum + tree[rr(pos)].sum;
return ind;
}
}
}seg;
int main()
{
while (scanf("%d",&n)!=EOF)
{
seg.buildtree(1,n,1);
for (int i = 1; i <= n; i++)
{
scanf("%d %d",&pos[i],&val[i]);
}
for(int i = n; i >=1; i--)
{
int ps = seg.query(pos[i]+1,1);
ans[ps] = val[i];
}
for (int i = 1; i <= n; i++)
{
printf("%d",ans[i]);
if (i != n)
printf(" ");
}
printf("\n");
}
return 0;
}