题意:
有 n n n个人来排队,第 i i i 个人来的时候排在第 p [ i ] p[i] p[i] 个人的后面,每个人都有一个数字 v [ i ] v[i] v[i] ,求最后按队列顺序的 v [ i ] v[i] v[i] 。
题解:
考虑倒着来,最后一个人的位置必定是确定的,即 p [ n ] + 1 p[n]+1 p[n]+1 。那么倒数第二个人呢?
这么想,第 n − 1 n-1 n−1来的时候,第 n n n 个人此时不在队列中,那么队列的状况就是去掉 p [ n ] + 1 p[n]+1 p[n]+1这个位置,剩下 n − 1 n-1 n−1个位置,那他的位置就是这 n − 1 n-1 n−1中的第 p [ n − 1 ] + 1 p[n-1]+1 p[n−1]+1个,即在剩下的 n − 1 n-1 n−1个空位中的第 p [ n − 1 ] + 1 p[n-1]+1 p[n−1]+1个。以此类推。
我们只需用线段树维护区间还剩下多少位置没被选,就可以查询出第 p [ i ] + 1 p[i]+1 p[i]+1 个空位在哪。
代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int ,int > pii;
const int mod=1e9+7;
const int MAXN=2e5+5;
const int inf=0x3f3f3f3f;
int p[MAXN],v[MAXN];
int ans[MAXN];
struct Node
{
int l,r;
int sum;
}node[MAXN<<2];
void build(int l,int r,int num)
{
node[num].l=l;
node[num].r=r;
if(l==r)
{
node[num].sum=1;
return;
}
int mid=(l+r)>>1;
build(l,mid,num<<1);
build(mid+1,r,num<<1|1);
node[num].sum=node[num<<1].sum+node[num<<1|1].sum;
}
void updata(int pos,int num)
{
if(node[num].l==node[num].r)
{
node[num].sum--;
return;
}
int mid=(node[num].l+node[num].r)>>1;
if(pos<=mid) updata(pos,num<<1);
else updata(pos,num<<1|1);
node[num].sum=node[num<<1].sum+node[num<<1|1].sum;
}
int query(int pos,int num)
{
if(node[num].l==node[num].r) return node[num].l;
int mid=(node[num].l+node[num].r)>>1;
if(node[num<<1].sum>=pos)
{
return query(pos,num<<1);
}
else return query(pos-node[num<<1].sum,num<<1|1);
}
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;i++)
{
scanf("%d%d",&p[i],&v[i]);
}
build(1,n,1);
for(int i=n;i>=1;i--)
{
int pos=query(p[i]+1,1);
ans[pos]=v[i];
updata(pos,1);
}
for(int i=1;i<=n;i++) printf("%d ",ans[i]);
printf("\n");
}
}