UVA 501 - Black Box(二分 + 树状数组 + 离散化)

题目链接:点击打开链接

思路: 该题虽然是求区间第k小, 但是实际上是不需要用到复杂的数据结构的, 我们只需要用树状数组来维护就行了, 因为以值作为树状数组下标, 那么这就是一个天然的单调区间, 用二分来查找就行了。  另外就是要离散化一下。

细节参见代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <string>
#include <vector>
#include <stack>
#include <bitset>
#include <cstdlib>
#include <cmath>
#include <set>
#include <list>
#include <deque>
#include <map>
#include <queue>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
typedef long double ld;
const ld eps = 1e-9, PI = 3.1415926535897932384626433832795;
const int mod = 1000000000 + 7;
const int INF = 0x3f3f3f3f;
// & 0x7FFFFFFF
const int seed = 131;
const ll INF64 = ll(1e18);
const int maxn = 30000 + 10;
int T,n,m,bit[maxn],id[maxn],a[maxn],b[maxn];
int sum(int x) {
    int ans = 0;
    while(x > 0) {
        ans += bit[x];
        x -= x & -x;
    }
    return ans;
}
void add(int x, int d) {
    while(x <= n) {
        bit[x] += d;
        x += x & -x;
    }
}
int main() {
    scanf("%d",&T);
    while(T--) {
        int cnt = 0;
        scanf("%d%d",&n,&m);
        for(int i = 1; i <= n; i++) {
            scanf("%d",&a[i]);
            id[cnt++] = a[i];
        }
        //  离散化
        sort(id, id+cnt);
        cnt = unique(id, id+cnt) - id;
        for(int j = 1; j <= m; j++) scanf("%d",&b[j]);
        // 初始化
        memset(bit, 0, sizeof(bit));
        int cur = 0;
        for(int i = 1; i <= m; i++) {
            // 加入元素的id
            if(b[i] > cur) {
                for(int j = cur+1; j <= b[i]; j++) {
                    int v = lower_bound(id, id+cnt, a[j]) - id + 1;
                    add(v, 1);
                }
                cur = b[i];
            }
            int l = 1, r = cnt;
            while(r > l) {
                int mid = (l + r) >> 1;
                int cur = sum(mid);
                if(cur >= i) r = mid;
                else l = mid + 1;
            }
            printf("%d\n",id[l-1]);
        }
        if(T) printf("\n");
    }
    return 0;
}


你可能感兴趣的:(数据结构,uva)