笔试刷题-百度

最新推荐文章于 2024-06-11 23:04:55 发布

m0_37794421

最新推荐文章于 2024-06-11 23:04:55 发布

阅读量137

点赞数

CC 4.0 BY-SA版权

分类专栏：笔试刷题文章标签：滑动窗口

本文链接：https://blog.youkuaiyun.com/m0_37794421/article/details/80768570

笔试刷题专栏收录该内容

81 篇文章

订阅专栏

本文介绍了一种使用双向队列实现滑动窗口的方法来解决C市向D市转移罪犯的问题，确保转移的罪犯数量固定且罪行总值不超过限定值。通过此算法可以高效地计算出所有符合条件的转移方案。

题目描述：

/**
C市现在要转移一批罪犯到D市，C市有n名罪犯，
按照入狱时间有顺序，另外每个罪犯有一个罪行值，值越大罪越重。
现在为了方便管理，市长决定转移入狱时间连续的c名犯人，
同时要求转移犯人的罪行值之和不超过t，问有多少种选择的方式？

输入描述:
第一行数据三个整数:n，t，c(1≤n≤2e5,0≤t≤1e9,1≤c≤n)，
第二行按入狱时间给出每个犯人的罪行值ai(0≤ai≤1e9)


输出描述:
一行输出答案。

输入例子1:
3 100 2
1 2 3

输出例子1:
2
*/

思路如下：

采用双向队列维护一个c大小滑动窗口即可
并更新即可计数即可

代码如下：

#include<stdio.h>
#include<iostream>
#include<deque>

#define MAX_N 200005

using namespace std;

int arr[MAX_N];

int main()
{
    int N, T, C;

    while(scanf("%d%d%d", &N, &T, &C)==3)
    {
        deque<int> que;
        for(int i=0; i<N; i++)
            scanf("%d", arr+i);
        int acc=0, res=0;
        //C>0 init que.size()==0
        for(int i=0; i<N; i++)
        {
            if(que.size()<C)
            {
                que.push_back(arr[i]);
                acc+=arr[i];
            }
            else if(que.size()==C)
            {
                int head=que.front();
                que.pop_front();
                acc-=head;
                que.push_back(arr[i]);
                acc+=arr[i];
            }
            if(acc<=T && que.size()==C)
            {
                res++;
            }
            else if(acc>T)
            {
                while(!que.empty() && acc>T)
                {
                    int head=que.front();
                    que.pop_front();
                    acc-=head;
                }
            }
        }
        printf("%d\n", res);
    }
    return 0;
}