【思维+树状数组】ZOJ-4117 BaoBao Loves Reading

最新推荐文章于 2025-05-08 14:59:06 发布

原创最新推荐文章于 2025-05-08 14:59:06 发布 · 694 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

树状数组专栏收录该内容

7 篇文章

订阅专栏

该博客介绍了ZOJ-4117 BaoBao Loves Reading问题，涉及操作系统中的FIFO调度算法和使用树状数组解决的方法。题目要求在书架上处理N本书，书桌仅能容纳K本书，计算不同K值下取书的次数。通过树状数组可以维护区间内不同书的数量，从而求得答案。

题目链接：传送门

参考博客：2019山东省赛 zoj 4117 E BaoBao Loves Reading 思维+树状数组

【题意】：

如果你学过操作系统，其实你就知道这个就是经典的调度算法的FIFO，先进先出算法。

如果你没有学过操作系统，这个就是说有N本书在书架，然后书桌只能容纳K本书，然后先进先出，请问当K取值为[1,n]时，每次拿书的次数为多少。

【题解】：

参考博客小喵的做法，其实只要知道结论真的很简单。

用树状数组维护一个东西，就是相同数字之间的距离，这个距离中有多少个数字是不同的，用Sum数组为存放。

树状数组维护：区间中种类的个数。

Sum[i]：K=i时，不操作的次数。

Ans = n - Sum[i]

然后这个K=i时，会为[i+1,n]作贡献

#include<cstdio>
#include<cstring>
using namespace std;
const int N = 1e5+100;
int c[N],a[N],sum[N],pre[N];
int lowbit(int x){
    return x&-x;
}
void update(int x,int val){
    while( x < N ){
        c[x] += val;
        x += lowbit(x);
    }
}
int getSum(int x){
    int Res = 0 ;
    while ( x > 0 ){
        Res += c[x];
        x -= lowbit(x);
    }
    return Res;
}
void Init(){
    memset(c,0,sizeof(c));
    memset(sum,0,sizeof(sum));
    memset(pre,0,sizeof(pre));
}
int main()
{
    int n,T;
    scanf("%d",&T);
    while(T--){
        Init();
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            if( pre[a[i]] ) {
                update( i , 1 );
                update( pre[a[i]], -1);
                int L = getSum(i) - getSum(pre[a[i]]-1);
                sum[L] ++ ;
                pre[a[i]] = i;
            }else{
                update(i,1);
                pre[a[i]] = i;
            }
        }
        for(int i=1;i<=n;i++){
            sum[i] = sum[i-1] + sum[i];
        }
        for(int i=1;i<=n;i++){
            printf("%d%c",n-sum[i],i==n?'\n':' ');
        }
    }
    return 0;
}