题意:给出n个人的信息,包括这个人来的时候站在第几个人的后面和这个人的数值,求所有人都到之后的序列。
解法:线段树单点更新。线段树结点表示当前区间里空位个数,倒序遍历,对于每个人来的时候站的位置a就是序列中空位的序号,每次插入的时候看左儿子的空位数,如果比a大说明a在左区间里,如果比a小说明在右区间里。
代码:
#include<stdio.h> #include<iostream> #include<algorithm> #include<string> #include<string.h> #include<math.h> #include<limits.h> #include<time.h> #include<stdlib.h> #include<map> #include<queue> #include<set> #include<stack> #include<vector> #include<iomanip> #define LL long long #define lson l, m, rt << 1 #define rson m + 1, r, rt << 1 | 1 using namespace std; struct node { int a, b; }p[200005]; int ans[200005];//记录答案序列 int sum[200005 << 2]; void pushup(int rt) { sum[rt] = sum[rt << 1] + sum[rt << 1 | 1]; } void build(int l, int r, int rt) { if(l == r) { sum[rt] = 1; return ; } int m = (l + r) >> 1; build(lson); build(rson); pushup(rt); } void update(node x, int l, int r, int rt) { if(l == r) { sum[rt] = 0; ans[l] = x.b;//找到这个人的位置后记录答案 return ; } int m = (l + r) >> 1; if(x.a + 1 <= sum[rt << 1])//在左儿子里 update(x, lson); else//在右儿子里 { x.a -= sum[rt << 1];//去掉在左儿子里的空位数 update(x, rson); } pushup(rt); } int main() { int n; while(~scanf("%d", &n)) { for(int i = 0; i < n; i++) { scanf("%d%d", &p[i].a, &p[i].b); } build(1, n, 1); for(int i = n - 1; i >= 0; i--) { update(p[i], 1, n, 1); } for(int i = 1; i <= n; i++) { if(i != 1) printf(" "); printf("%d", ans[i]); } printf(" "); } return 0; }
不知道为啥交G++就会T……交C++1s+