ZOJ 4117 BaoBao Loves Reading (树状数组前缀和)

最新推荐文章于 2022-03-17 21:43:34 发布

原创最新推荐文章于 2022-03-17 21:43:34 发布 · 315 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构

数据结构同时被 2 个专栏收录

25 篇文章

订阅专栏

OJ基础题

13 篇文章

订阅专栏

博客给出题目链接，介绍利用树状数组维护书桌整个时间段的书本数目，通过区间查询得到出现相同书本时的时间差，此时间差即为可省去的去书柜拿书次数，书桌容量等于时间差，最后用前缀和得到容量1到n的可省次数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=6019

用树状数组维护书桌整个时间段的书本数目，区间查询得到出现相同书本时的时间差，即为可以省去的去书柜拿书次数(书桌容量为即时间差)，最后利用前缀和，得到容量1~n的可省次数

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <cstring>
#include <queue>
#define eps 1e-8
using namespace std;
typedef long long ll;
static const int MAX_N = 1e5 + 5;
static const ll Mod = 233;
static const int INF = 0x3f3f3f3f;
int C[MAX_N], pr[MAX_N], rev[MAX_N];
int lowbit(int x){
    return x & (-x);
}
void add(int x, int y, int n){
    for(int i = x; i <= n; i += lowbit(i))C[i] += y;
}
int get_sum(int x, int y){
    int ret1 = 0, ret2 = 0;
    for(int i = y; i; i -= lowbit(i)) ret2 += C[i];
    for(int i = x; i; i -= lowbit(i)) ret1 += C[i];
    return ret2 - ret1;
}
int main(){
//    freopen("input.txt", "r", stdin);
//    freopen("output.txt", "w", stdout);
    int T;
    scanf("%d", &T);
    while(T--){
        int n;
        scanf("%d", &n);
        for(int i = 0; i <= n; ++i){
            C[i] = 0;
            pr[i] = 0;
            rev[i] = 0;
        }
        for(int i = 1; i <= n; ++i){
            int v;
            scanf("%d", &v);
            if(pr[v]){
                add(pr[v], -1, n);
                add(i, 1, n);
                int p = get_sum(pr[v] - 1, i);    
                ++rev[p];    //容量为p时可省次数+1
            }
            else add(i, 1, n);
            pr[v] = i;
        }
        for(int i = 1; i <= n; ++i) rev[i] += rev[i - 1];    //前缀和,前面能省的，后面一定能省
        for(int i = 1; i <= n; ++i) printf("%d%c", n - rev[i], i == n ? '\n' : ' ');
    }
}