POJ 2182 Lost Cows (树状数组 && 二分查找)

题意:给出数n, 代表有多少头牛, 这些牛的编号为1~n, 再给出含有n-1个数的序列, 每个序列的数 ai 代表前面还有多少头比 ai 编号要小的牛, 叫你根据上述信息还原出原始的牛的编号序列

分析:如果倒着看这个序列的话, 那序列的最后一个元素就能够确定一个编号。举个例子:如果序列的最后一个元素为0, 那就说明这头牛前面再也没有比它编号更小的牛了, 所以这头牛的编号肯定是最大的, 我们只要给它所在的编号加个标记, 然后继续根据倒数第二个、第三个……来依次确定便可还原整个序列, 这里可以使用树状数组做, 初始化全部加1操作, 然后开始枚举编号, 看哪个编号前面是有多少比其编号小的牛, 即区间求和, 一旦和一开始给出的序列元素相同则确定是此编号, 确定一头之后便抹去这头牛的编号, 即add(num, -1), 时时更新即可, 但是这里如果用for循环去枚举和的话未免太慢, 但也能AC, 这里考虑使用二分查找便很快了!

瞎搞:这题实际上还可以用线段树做, 做法大同小异, 但没想到的是, 这还是一个DP可以解决的题目……

树状数组:

#include<stdio.h>
#include<string.h>
#include<string>
#include<iostream>
#include<algorithm>
#include<stack>
#define lowbit(i) (i&(-i))
#define LL long long
using namespace std;
int c[8001];
int ans[8001];
int n;
inline void add(int i, int val)
{
    while(i<=n){
        c[i] += val;
        i += lowbit(i);
    }
}
int sum(int i)
{
    int ans = 0;
    while(i>0){
        ans += c[i];
        i -= lowbit(i);
    }
    return ans;
}
int Bin_search(int L, int R, int key)
{
    int mid;
    while(L < R){
        mid = L + ((R-L)>>1);
        if(sum(mid)-1 < key) L = mid+1;
        else R = mid;
    }
    return R;
}
int main(void)
{
    scanf("%d", &n);
    memset(c, 0, sizeof(c));
    stack<int> st;
    st.push(0);
    for(int i=1; i<=n-1; i++){
        int temp;
        scanf("%d", &temp);
        st.push(temp);
        add(i, 1);
    }
    add(n, 1);
    int top = 0;
    while(!st.empty()){
        int temp = st.top();
        st.pop();
        int num = Bin_search(1, n, temp);
        ans[top++] = num;
        add(num, -1);
    }
    for(int i=top-1; i>=0; i--){
        printf("%d\n", ans[i]);
    }
    return 0;
}

线段树:

#include<iostream>
#include<string.h>
#include<stdio.h>
using namespace std;
typedef struct segment
{
    int L, R, len;
}T;///线段树类型
int a[10000], result[10000];///a储存input,result储存output
T tree[30000];///线段树数组
///------------------------------------
void init(int, int, int);
int query(int, int);
///------------------------------------
int main(void)
{
    std::ios::sync_with_stdio(false);///关闭同步,加快读入速度
    int n;
    cin>>n;
    a[1] = 0;///第一个前面不可能有比它小的数
    for(int i=2; i<=n; i++){
        cin>>a[i];
    }
    init(1, 1, n);///(root, L, R)
    for(int i=n; i>=1; i--){
        result[i] = query(1, a[i]+1);
    }
    for(int i=1; i<=n; i++) cout<<result[i]<<endl;
    return 0;
}
///--------------------------------------------
void init(int root, int L, int R)///给线段树初始化从1~n的数据
{
    tree[root].L = L;
    tree[root].R = R;
    tree[root].len = R-L+1;
    if(L == R) return ;
    init(2*root, L, (L+R)/2);
    init(2*root+1, (L+R)/2+1, R);
}
///--------------------------------------------
int query(int node, int aim)
{
    tree[node].len--;
    if(tree[node].L == tree[node].R) return tree[node].L;
    if(aim <= tree[2*node].len) return query(node*2, aim);
    else return query(node*2+1, aim-tree[node*2].len);
}

时间: 2024-10-29 10:46:14

POJ 2182 Lost Cows (树状数组 && 二分查找)的相关文章

POJ 2182 Lost Cows (树状数组)

Lost Cows Time Limit: 1000MS   Memory Limit: 65536K Total Submissions: 9660   Accepted: 6219 Description N (2 <= N <= 8,000) cows have unique brands in the range 1..N. In a spectacular display of poor judgment, they visited the neighborhood 'waterin

POJ 2892 Tunnel Warfare (树状数组+二分)

题目大意: 三个操作 D pos  将pos位置摧毁,让它和周围不相连. Q pos 问和pos 相连的有多少个村庄. R 修复最近摧毁的村庄. 思路分析: 树状数组记录这个区间有多少个1. 如果  [s-e] 有e-s+1个1 的话.那么这个区间是相连的. 这样的话,我们就可以用二分的办法求出与某个位置最大相连的数量. 还有这里二分 while(l<=r) { if(满足) { ans=mid; l=mid+1; } else r=mid-1; } #include <cstdio>

poj2182Lost Cows——树状数组快速查找

题目:http://poj.org/problem?id=2182 从后往前确定,自己位置之前没有被确定的且比自己编号小的个数+1即为自己的编号: 利用树状数组快速查找,可另外开一个b数组,角标为编号大小,而其值为是否使用,二分查找到恰好满足条件的位置,向后一直找到没被用过的第一个编号即为此位置编号. 代码如下: #include<iostream> #include<cstdio> using namespace std; int n,a[8005],f[8005],ans[80

POJ 2828 Buy Tickets (线段树 or 树状数组+二分)

题目链接:http://poj.org/problem?id=2828 题意就是给你n个人,然后每个人按顺序插队,问你最终的顺序是怎么样的. 反过来做就很容易了,从最后一个人开始推,最后一个人位置很容易就确定了,那最后第二个人的位置也可以推(与最后一个人的位置无关)...依次就都可以确定所有的人了. 用前缀和的思想,要是这个人的位置确定了,那么就标记这个人位置的值为0,然后回溯更新,跟求逆序对个数的思想比较类似. 线段树: 1 #include <iostream> 2 #include &l

POJ 2481 Cows(树状数组)

Description Farmer John's cows have discovered that the clover growing along the ridge of the hill (which we can think of as a one-dimensional number line) in his field is particularly good. Farmer John has N cows (we number the cows from 1 to N). Ea

HDU 2852 KiKi&#39;s K-Number (树状数组 &amp;&amp; 二分)

题意:给出对容器的总操作次数n, 接下来是这n个操作.这里对于一个容器提供三种操作, 分别是插入.删除和查找.输入0  e表示插入e.输入1  e表示删除e,若元素不存在输出No Elment!.输入2  e  k表示查找比e大且第k大的数, 若不存在则输出Not Find! 分析:这里考虑树状数组做的原因是在第三个操作的时候, 只要我们记录了元素的总数, 那通过求和操作, 便能够高效地知道到底有多少个数比现在求和的这个数要大, 例如 tot - sum(3)就能知道整个集合里面比3大的数到底有

hdu 1541/poj 2352:Stars(树状数组,经典题)

Stars Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)Total Submission(s): 4052    Accepted Submission(s): 1592 Problem Description Astronomers often examine star maps where stars are represented by points on a plan

ZOJ 3635 树状数组+二分

这题那时怎么想就是想不出来--而且今晚没有多大状态,自己都晕了--一题没做出来-- baoge解释好久才懂--唉--线段树,树状数组用得还是不够熟啊-- WA了二发,才知道二分错了,二分好久不用,老是出错了现在-- #include<iostream> #include<cstring> #include<string> #include<cstdio> #define sca(a) scanf("%d",&a) #define

HDU 5592 ZYB&#39;s Premutation(树状数组+二分)

题意:给一个排列的每个前缀区间的逆序对数,让还原 原序列. 思路:考虑逆序对的意思,对于k = f[i] - f[i -1],就表示在第i个位置前面有k个比当前位置大的数,那么也就是:除了i后面的数字之外,它是在剩下的数字当中第k+1大的. 知道这个之后,可以用树状数组来帮助找出剩下的数中第k大的数,刚开始我们可以让1-n中每个元素都标记为1,那么他们的前缀和就代表它是第几小.所以,我们可以对于他们的和来二分快速寻找第k大数.其实在树状数组里面是按照第(i-k)小来找的.找完之后要删除这个元素的