ZOJ 3635 Cinema in Akiba（二分+树状数组）

解决座位分配问题的经典水题

最新推荐文章于 2018-09-03 21:37:01 发布

原创最新推荐文章于 2018-09-03 21:37:01 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#zoj #二分 #树状数组

ACM竞赛同时被 3 个专栏收录

217 篇文章

订阅专栏

高效算法

71 篇文章

订阅专栏

线段树和树状数组

60 篇文章

订阅专栏

本文探讨了一个经典的计算机科学问题——n个人轮流选择座位，第i个人选择第a[i]个空座的问题。通过使用二分搜索和树状数组等高效数据结构，我们能够快速找到每个人最终的座位情况。该文提供了详细的算法实现步骤和代码，帮助读者理解并解决此类问题。

题目链接：点击打开链接

题意：n个人轮流做到座位上，第i个人做到第a[i]个空座上，求最终每个人的座位情况。

思路：经典水题，二分套树状数组。

细节参见代码：

#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 = int(1e9);
// & 0x7FFFFFFF
const int seed = 131;
const ll INF64 = ll(1e18);
const int maxn = 50000 + 10;
int T,n,m,v,q,ans[maxn],bit[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() {
    while(~scanf("%d",&n)) {
        memset(bit, 0, sizeof(bit));
        for(int i=1;i<=n;i++) {
            scanf("%d",&v);
            int mid, l = 1, r = n;
            while(r > l) {
                mid = (l + r) >> 1;
                int res = mid - sum(mid);
                if(res >= v) r = mid;
                else l = mid + 1;
            }
            add(l, 1);
            ans[i] = l;
        }
        scanf("%d",&q);
        bool ok = true;
        while(q--) {
            scanf("%d",&v);
            if(ok) printf("%d",ans[v]);
            else printf(" %d",ans[v]);
            ok = false;
        }
        printf("\n");
    }
    return 0;
}