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 n;
inline void add(int i, int val)
{
    while(i<=n){
        c[i] += val;
        i += lowbit(i);
    }
}
int sum(int i)
{
    ;
    ){
        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)>>);
         < key) L = mid+;
        else R = mid;
    }
    return R;
}
int main(void)
{
    scanf("%d", &n);
    memset(c, , sizeof(c));
    stack<int> st;
    st.push();
    ; i<=n-; i++){
        int temp;
        scanf("%d", &temp);
        st.push(temp);
        add(i, );
    }
    add(n, );
    ;
    while(!st.empty()){
        int temp = st.top();
        st.pop();
        , n, temp);
        ans[top++] = num;
        add(num, -);
    }
    ; i>=; i--){
        printf("%d\n", ans[i]);
    }
    ;
}

线段树:

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