POJ 2828 Buy Tickets(树状数组)
最后更新于:2022-04-01 15:53:09
题目链接:[点击打开链接](http://poj.org/problem?id=2828)
题意:给n个人依次插队的情况, 要求求出最终的队伍情况。
该题可以用树状数组很方便的维护。
如果从前向后扫的话, 每次插队都会影响这个人后面的情况, 所以我们可以倒着进行, 那么最后一个人的位置就是最终位置, 对于前面的人的位置, 只受后面人的影响, 所以等价于其最终位置是第pos[i]+1个空位。 那么也就是说要快速的找到第pos[i]+1个空位。 显然, 这又可以用二分位置和树状数组维护的方法来优化算法。
细节参见代码:
~~~
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
const double PI = acos(-1.0);
const double eps = 1e-6;
const int INF = 1000000000;
const int maxn = 200000 + 10;
int T,n,m,bit[maxn],ans[maxn];
struct node {
int pre, v;
}a[maxn];
int sum(int x) {
int ans = 0;
while(x > 0) {
ans += bit[x];
x -= x & -x;
}
return ans;
}
void add(int x, int d) {
while(x <= n) {
bit[x] += d;
x += x & -x;
}
}
int main() {
while(~scanf("%d",&n)) {
memset(bit, 0, (n+1)*sizeof(bit[0]));
for(int i=1;i<=n;i++) scanf("%d%d",&a[i].pre,&a[i].v);
for(int i=n;i>=1;i--) {
int l = 1, r = n, mid;
while(r > l) {
mid = (l + r) >> 1;
if(mid - sum(mid) > a[i].pre) r = mid;
else l = mid + 1;
}
ans[l] = a[i].v;
add(l, 1);
}
printf("%d",ans[1]);
for(int i=2;i<=n;i++) printf(" %d",ans[i]);
printf("\n");
}
return 0;
}
~~~