POJ 2182 Lost Cows(树状数组+二分)

题目点我点我点我


题目大意:有n头牛,给出第2~n头牛在它前面有多少头牛号码比自己小的,输出每头牛的号码。


解题思路:从后往前看,最后一头牛可以确定它的号码,因为知道了前面有k头比它小的,k+1即为它号码。记a[i]为第i头牛在它前面有多少头牛号码比它小的(即输入值),

c为它后面有多少头牛号码比它小的,本身号码为ans,则有a[i]+c=ans-1。

变形可得:ans-1-x=a[i]。其中,x用树状数组来维护。

ans用二分查找来确定,当ans-1-x<a[i]时,L=mid+1,ans-1-x>=a[i]时,R=mid。

那么这里问题来了,为什么不直接让ans=mid的呢?因为直接等于的话样例有可能答案为

1

3

5

3

1

明显这个答案是错的,即在从后往前推导过程中有可能后面的值已经出现过了前面继续用这个值,故需要排除掉后面出现过的值。


/* ***********************************************
┆  ┏┓   ┏┓ ┆
┆┏┛┻━━━┛┻┓ ┆
┆┃       ┃ ┆
┆┃   ━   ┃ ┆
┆┃ ┳┛ ┗┳ ┃ ┆
┆┃       ┃ ┆
┆┃   ┻   ┃ ┆
┆┗━┓ 马 ┏━┛ ┆
┆  ┃ 勒 ┃  ┆      
┆  ┃ 戈 ┗━━━┓ ┆
┆  ┃ 壁     ┣┓┆
┆  ┃ 的草泥马  ┏┛┆
┆  ┗┓┓┏━┳┓┏┛ ┆
┆   ┃┫┫ ┃┫┫ ┆
┆   ┗┻┛ ┗┻┛ ┆
************************************************ */

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
using namespace std;

#define rep(i,a,b) for (int i=(a),_ed=(b);i<=_ed;i++)
#define per(i,a,b) for (int i=(b),_ed=(a);i>=_ed;i--)
#define pb push_back
#define mp make_pair
const int inf_int = 2e9;
const long long inf_ll = 2e18;
#define inf_add 0x3f3f3f3f
#define mod 1000000007
#define LL long long
#define ULL unsigned long long
#define MS0(X) memset((X), 0, sizeof((X)))
#define SelfType int
SelfType Gcd(SelfType p,SelfType q){return q==0?p:Gcd(q,p%q);}
SelfType Pow(SelfType p,SelfType q){SelfType ans=1;while(q){if(q&1)ans=ans*p;p=p*p;q>>=1;}return ans;}
#define Sd(X) int (X); scanf("%d", &X)
#define Sdd(X, Y) int X, Y; scanf("%d%d", &X, &Y)
#define Sddd(X, Y, Z) int X, Y, Z; scanf("%d%d%d", &X, &Y, &Z)
inline int read(){int ra,fh;char rx;rx=getchar(),ra=0,fh=1;while((rx<'0'||rx>'9')&&rx!='-')rx=getchar();if(rx=='-')fh=-1,rx=getchar();while(rx>='0'&&rx<='9')ra*=10,ra+=rx-48,rx=getchar();return ra*fh;}
//#pragma comment(linker, "/STACK:102400000,102400000")

int n;
int c[10000];
int a[10000];
int ans [10000];

int lowbit(int x)
{
    return x & (-x);
}

void update(int i,int val)
{
    while(i<=n)
    {
        c[i] += val;
        i += lowbit(i);
    }
}

int sum(int i)
{
    int ans = 0;
    while(i)
    {
        ans += c[i];
        i -= lowbit(i);
    }
    return ans;
}

int binary(int x)
{
    int l = 1, r = n,mid;
    while(l<r)
    {
        mid = (l+r) >> 1;
        int t = sum(mid);
        if(mid-1-t<x)l = mid + 1;
        else r = mid;
    }
    return r;
}

int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	ios::sync_with_stdio(0);
	cin.tie(0);
	n = read();
	for(int i=1;i<n;i++)a[i] = read();
	for(int i=n-1;i>=0;i--)
    {
        int v = binary(a[i]);
        ans[i] = v;
        update(v,1);
    }
	for(int i=0;i<n;i++)printf("%d\n",ans[i]);


	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值