POJ 2182 Lost Cows(线段树，单点更新)

最新推荐文章于 2022-09-03 21:45:05 发布

原创最新推荐文章于 2022-09-03 21:45:05 发布 · 279 阅读

0 ·

CC 4.0 BY-SA版权

其他专栏收录该内容

65 篇文章

订阅专栏

题意：    有N头牛，编号为1~N ，乱序排成一列，现在已知每头牛前面有多少头牛比他的编号小，求排队后从前往后数，输出每头牛的编号

注意：  第一头前面没有牛 所以初始值为0

思路： 最后一头牛为 a  则 牛的序列为 a+1  去掉编号 a+1 再从后往前推 （其实这题数据很水 暴力也能过  暴力出奇迹 2333）

AC 代码;

#include <iostream>

#include <cstdio>
using namespace std;
#define N 2000000
int n, a[N + 5];
int ans[N + 5];
struct tree
{
    int l;
    int r;
    int v;
} tree[N * 4];
void build(int id, int l, int r)
{
    tree[id].l = l;
    tree[id].r = r;
    tree[id].v = r - l + 1;
    if (r == l)
    {
        return;
    }
    int mid = (l + r) / 2;
    build(id * 2, l, mid);
    build(id * 2 + 1, mid + 1, r);
}
int querry(int id, int k)
{
    tree[id].v--; //区间长度减小 表示这个区间内的点背删除了一个
    if (tree[id].l == tree[id].r)
        return tree[id].l;
    if (k <= tree[id * 2].v)
        return querry(id * 2, k);
    else
        return querry(id * 2 + 1, k - tree[id * 2].v);
}
int main()
{
    int i;
    while (scanf("%d", &n) != EOF)
    {
        for (i = 2; i <= n; i++)
            scanf("%d", &a[i]);
        a[1] = 0;
//第一头牛前面没有 所以初始化为0 build(1, 1, n); for (i = n; i >= 1; i--)
        {
            ans[i] = querry(1, a[i] + 1);
        }
        for (i = 1; i <= n; i++)
        {
            printf("%d\n", ans[i]);
        }
    }
    return 0;
}