罪犯转移

题目描述

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

输入描述:

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

输出描述:

一行输出答案。

输入例子:

3 100 2
1 2 3

输出例子:2

思路：其实题目讲得很复杂，但要做的事情很简单，就是连续c个罪犯的罪犯值相加不超过t。比如有5人，连续3人，这里的组合就只有1 2 3 、2 3 4 、 3 4 5这几种。所以写循环，测相加值就可以了。

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

int plus(int *in,int start,int end)
    {
    int sum=0;
    for(int i=start;i<end;i++)
        sum = sum+in[i];
    return sum;
    
}


int main()
    {
    int n=0,t=0,c=0;
    int wayscount=0;
    int *crime;
    while(scanf("%d%d%d",&n,&t,&c) != EOF)
        {
        if(c<=n)
            {
                wayscount=0;
                sum=0;
        	crime = (int *)malloc(sizeof(int)*n);
        	for(int i=0;i<n;i++)
            	scanf("%d",&crime[i]);
            
            int start = 0;
            while(start+c<=n)
                {
                if(plus(crime,start,start+c)<t)
                    start++;
                    wayscount++;
            }
            printf("%d\n",wayscount);
            free(crime);
        }
    }
    
    return 0;
}




这是我一开始的代码，结果是运行时间过长，无法编译通过。这就很尴尬了。我这里都没有很大量的循环。

然后我就去参考别人的代码。发现问题出在，我每次都在将c个罪犯值重新加一次，所以在数据量大的时候浪费了很多时间。

这里参考了别人的思路，感觉很有用，就记下来。

这里类似一个滑块窗口。

这样一来累加就只做一次，大大缩减了运行时间。我这里为了方便，就没有将累加函数写回main里了。如果对时间要求很高，建议全写在main里，毕竟函数跳转，变量压栈，出栈这些也都是需要时间的。贴上改进后的代码

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

int plus(int *in,int start,int end)
    {
    int sum=0;
    for(int i=start;i<end;i++)
        sum = sum+in[i];
    return sum;
    
}


int main()
    {
    int n=0,t=0,c=0;
    int wayscount=0;
    int *crime;
    int sum=0;
    while(scanf("%d%d%d",&n,&t,&c) != EOF)
        {
        if(c<=n)
            {
            wayscount = 0;
            sum = 0;
        	crime = (int *)malloc(sizeof(int)*n);
        	for(int i=0;i<n;i++)
            	scanf("%d",&crime[i]);
            
            int start = 0;
            sum = plus(crime,start,start+c);
            if(sum<=t)
                wayscount++;
            for(int j=c;j<n;j++)
                {
                sum=sum-crime[j-c];
                sum=sum+crime[j];
                if(sum<=t) wayscount++;
            }
            
            printf("%d\n",wayscount);
            free(crime);
        }
    }
    
    return 0;
}