复习线段树时看到了这种把线段树定义成结构体的写法,优点确实很明显,很容易理解线段树的结构,参考了别人代码第一次写出这种优美的线段树,争取掌握吧。
题意:有n个人,给出n个人要插入的位置与其价值,输出最后的价值,从最后一个插入的人开始逆推出所有人想要插入的队列并对线段树进行更新即可
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include <cmath>
#include <iomanip>
#include <cstdlib>
using namespace std;
const int maxn=1100000;
int n;
int ans[maxn];
struct node
{
int n,val;
}a[maxn];
struct tree
{
int l;
int r;
int n;//区间
}b[maxn];
void init()
{
int k=1;
while(k<n)
{
k*=2;
}
//cout<<k<<endl;
for(int i=k;i<2*k;i++)//更新叶子节点
{
b[i].l=b[i].r=i-k+1;
b[i].n=1;//叶子节点只能放一个人
}
for(int i=k-1;i>0;i--)//更新非叶子节点
{
b[i].l=b[2*i].l;
b[i].r=b[2*i+1].r;
b[i].n=b[2*i].n+b[2*i+1].n;
}
}
void insert(int i,int x,int m)
{
if(b[i].l==b[i].r)//找到叶子结点并赋值
{
ans[b[i].l]=m;
b[i].n=0;
return ;
}
if(x<=b[2*i].n)
{
insert(2*i,x,m);
}
else
{
insert(2*i+1,x-b[2*i].n,m);
}
b[i].n--;
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)
{
scanf("%d%d",&a[i].n,&a[i].val);
}
init();
for(int i=n;i>0;i--)
{
insert(1,a[i].n+1,a[i].val);
}
printf("%d",ans[1]);
for(int i=2;i<=n;i++)
{
printf(" %d",ans[i]);
}
cout<<endl;
}
return 0;
}