罪犯转移

罪犯转移

题目描述

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

输出

2

多组数据

//先计算前c个数的累加值sum，之后将指针i指向数组下标c处，指针每前移一位，
//sum-=a[i-c]; sum+=a[i];使效率变为O(n)
#include <iostream>
using namespace std;
int weight[200005];
  
int main(){
    int n, t, c;
    while(cin >> n >> t >> c){
        int i, sum = 0, cnt = 0;
        for(i = 0; i < n; ++i)
            cin >> weight[i];
        if(c > n) continue;
        for(i = 0; i < c; ++i)
            sum += weight[i];
        if(sum <= t) ++cnt;
        for(i = c; i < n; ++i){
            sum -= weight[i-c];
            sum += weight[i];
            if(sum <= t) ++cnt;
        }
        cout << cnt << endl;
    }
    return 0;
}

运行超时：

#include <iostream>
using namespace std;
int n,t,c;
int a[200005];
int ans=0;
int main(){
    while(cin>>n>>t>>c){
        for(int i=1;i<=n;i++){
               cin>>a[i];
        }
        for(int i=1;i<=n-c+1;i++){
            int sum=0;
            for(int j=0;j<c;j++){
                sum+=a[i+j];
            }
            if(sum<=t) ans++;
        }
        cout<<ans<<endl;
    }
    return 0;
}