中医药院校程序设计竞赛备赛一-Problem E: k倍区间（抽屉原理）

Problem E: k倍区间

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 158  Solved: 48
[Submit][Status][Web Board]

Description

给定一个长度为N的数列，A1, A2, ... AN，如果其中一段连续的子序列Ai, Ai+1, ... Aj(i <= j)之和是K的倍数，我们就称这个区间[i, j]是K倍区间。 

你能求出数列中总共有多少个K倍区间吗？

 

Input

每个数据包含多组输入数据

第一行包含两个整数N和K。(1 <= N, K <= 100000)  
以下N行每行包含一个整数Ai。(1 <= Ai <= 100000)

 

Output

输出一个整数，代表K倍区间的数目。

 

Sample Input

5 2 1 2 3 4 5

Sample Output

6

补题中，代码来自同学...抽屉原理我OK，hash我OK，但是这道题我真的没做出来！dbq，我反思！我还是应该脚踏实地一点。

【通过代码】

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int sum[maxn],num[maxn];
int main()
{
    int n,k;
    while(~scanf("%d%d",&n,&k))
    {
        int cnt=0;
        memset(num,0,sizeof(num));
        memset(sum,0,sizeof(sum));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&num[i]);
            num[i]=(num[i]+num[i-1])%k;
            if(num[i]==0)cnt++;
        }
        //sum[0]=1;
        for(int i=1;i<=n;i++)
        {
             cnt+=sum[num[i]];
             sum[num[i]]++;
        }
        printf("%d\n",cnt);
    }
    return 0;
}

【比赛代码（TLE）】

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <vector>
#include <stack>
#include <set>
#include <algorithm>
using namespace std;
const int maxn = 1e5 + 5;
int sum[maxn],a[maxn];
int n,k;
void init()
{
    sum[0] = 0;
    for(int i = 1;i <= n;i ++)
        sum[i] = sum[i - 1] + a[i];
}
int main()
{
    
    while(scanf("%d%d",&n,&k) != EOF)
    {
        memset(sum,0,sizeof(sum));
        memset(a,0,sizeof(a));
        
        int cnt = 0;
        for(int i = 1;i <= n;i++)
            scanf("%d",&a[i]);
        init();
        for(int i = 1;i < n;i++)//这里两个循环显然超时的！！！
        {
            for(int j = i ;j <= n;j++)
            {
                if((sum[j] - sum[i - 1]) % k == 0)
                    cnt++;
            }
        }
        printf("cnt: %d\n",cnt); 
    }
    return 0;
}