蓝桥杯学习记录||1230. K倍区间

AcWing||1230. K倍区间

活动地址：https://www.acwing.com/activity/content/19/

考察要点：前缀和

题目要求

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

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

输入格式
第一行包含两个整数 N 和 K。

以下 N 行每行包含一个整数 Ai。

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

数据范围

1≤N,K≤100000,
1≤Ai≤100000

输入样例：

5 2
1
2
3
4
5

输出样例：

6

题目地址：https://www.acwing.com/problem/content/1232/

解析：一开始直接写个前缀和，然后枚举区间长度从0~n求出区间的和S % k查看是否是k倍区间，但是运行了两个数据就超时了。

只能对代码进行优化，已知
x % k = z;
y % k = z;
(y - x) % k = 0
这是求K倍区间的关键(z != 0)

1. 第一次求出 余数为z 是无法进行组合的需要记录下来cnt[z] = 1;
2. 第二次求出 余数为z 第二次的数可以组合第一次的数 找到
   （cnt[z] =1）1个 k倍区间，记录下(cnt[z] ++)
3. 第三次求出 余数为z第三次的数可以组合前面的数 找到
   (cnt[z] =2) 2个k倍区间 ，记录下(cnt[z] ++)
   （z == 0 是特殊情况 第一次求出 就可以是K倍区间 所以 cnt[0] 初始值就是 1）
   

#include<iostream>
 using namespace std;
 typedef long long LL;
 const int N = 1e5 + 10;
 int n, k;
 LL s[N];        //记录数据->前缀和
 int count[N];    //count[x] 为 s[i] % k = x  的个数
int main()
{
    cin >> n >> k;
    for (int i = 1; i <= n; i ++ )  //避免越界，从一开始
    {
        scanf("%d", &s[i]);    //输入数据
        s[i] += s[i - 1];      //计算前缀和
    }
    LL res = 0;
    count[0]++;     //余零找到第一个数就可加一，其他的需要两个及两个以上才有加
    for (int i = 1; i <= n; i ++ )
    {
        res += count[s[i] % k];
        count[s[i] % k] ++;
    }
    printf("%lld\n",res);
    return 0;
    
}

本题:1230. K倍区间
代码参考:视频讲解