K倍区间【详解版 -- 蓝桥杯】

名字想不称展

已于 2023-03-20 23:41:35 修改

阅读量1.8k

点赞数 20

分类专栏：算法刷题仓库文章标签：蓝桥杯算法职场和发展

于 2023-03-20 23:36:04 首次发布

本文链接：https://blog.csdn.net/m0_63613132/article/details/129678374

版权

算法刷题仓库专栏收录该内容

211 篇文章 1 订阅

订阅专栏

文章介绍了如何解决寻找数列中K倍区间的算法问题，提供了两种方法：暴力搜索和利用前缀和转换。通过前缀和数组的处理和对每个元素模K的余数进行计数，可以有效地计算出满足条件的区间数量。代码示例展示了具体的实现过程。

摘要由CSDN通过智能技术生成

😊😊 😊😊
不求点赞，只求耐心看完，指出您的疑惑和写的不好的地方，谢谢您。本人会及时更正感谢。希望看完后能帮助您理解算法的本质
😊😊 😊😊

目录

题目描述：
小白到进阶各种解法：
一、暴搜：😊
代码：

二、本题考察算法：前缀和转换😊
代码：

题目描述：

给定一个长度为N的数列，A1, A2, … AN，如果其中一段连续的子序列Ai, Ai+1, … Aj(i <= j)之和是K的倍数，我们就称这个区间[i, j]是K倍区间。
你能求出数列中总共有多少个K倍区间吗？

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

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

输入：

5 2 1 2 3 4 5

输出：

6

小白到进阶各种解法：

一、暴搜：😊

在这里插入图片描述

代码：

在这里插入图片描述

二、本题考察算法：前缀和转换😊

$a$ 为原数组， $s$ 为原数组的前缀和数组！
在这里插入图片描述

预处理出来 $a$ 数组的前缀和数组 $s$ ；

for (int i=1; i <= n; i ++)
    {
        scanf ("%d", &a[i]);
        s[i] = s[i-1] + a[i];
    }

对 $s$ 数组进行处理，从左往右将每个 $s[i]\%k$ 进行记录；方便后续判断！一边记录，一边累加。记住此时的问题转化成了与 $s[i]\%k$ 相等的 数 的个数，这些 数 必然在 $i$ 的左边之前，不仅出现了，还记录了 它%k 的余数，的个数+1；具体看代码：

    long long res=0;
    cnt[0] = 1;	后面有解释
    for (int i=1; i <= n; i ++)
    {
        res += cnt[s[i]%k];
        cnt[s[i]%k] ++; //记录值为 s[i]%k 的s[i]的个数+1；
    }

为什么会初始化： $c n t [0] = 1$ ？
$c n t$ 数组的含义是：假设现在有 3个结果，分别是 1，2，3。那么 $c n t$ 数组就负责记录的是： $s$ 数组的元素里 $\%k=1或者\%k=2 或者\%k=3，$ 的元素的个数各有多少个！
所以 $c n t [0] = 1$ 表示 $s$ 数组中 $\%k=0$ 的元素的个数。

总之就是为了避免遗漏这种情况，而我们的 $f or$ 循环根本递推不出来这种情况。

代码：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1e5 + 10;
int a[N];
long long s[N];
int n, k;
int cnt[N];

int main()
{
    scanf ("%d%d", &n, &k);
    
    for (int i=1; i <= n; i ++)
    {
        scanf ("%d", &a[i]);
        s[i] = s[i-1] + a[i];
    }
    
    long long res=0;
    cnt[0] = 1;
    for (int i=1; i <= n; i ++)
    {
        res += cnt[s[i]%k];
        cnt[s[i]%k] ++; //记录值为 s[i]%k 的s[i]的个数+1；
    }
    cout << res << endl;
    return 0;
}