k 倍区间

最新推荐文章于 2024-09-22 11:42:19 发布

大大胡萝卜

最新推荐文章于 2024-09-22 11:42:19 发布

阅读量80

点赞数

文章标签：算法蓝桥杯

本文链接：https://blog.csdn.net/weixin_51641169/article/details/129808174

版权

文章介绍了如何解决蓝桥杯2017年竞赛中的一道题目，该题目涉及寻找一个数列中所有连续子序列的和为K倍数的区间。通过使用前缀和优化，将初始的O(n^3)复杂度降低到O(n^2)，并进一步优化到线性时间复杂度，通过记录每个余数的数的个数来计算K倍区间数目。

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

[蓝桥杯 2017 省 B] k 倍区间

题目描述

给定一个长度为 $N$ 的数列， $A_1,A_2, \cdots A_N$ ，如果其中一段连续的子序列 $A_i,A_{i+1}, \cdots A_j(i \le j)$ 之和是 $K$ 的倍数，我们就称这个区间 $[i, j]$ 是 $K$ 倍区间。

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

输入格式

第一行包含两个整数 $N$ 和 $K$ $\le N,K \le 10^5)$ 。

以下 $N$ 行每行包含一个整数 $A_i$ $\le A_i \le 10^5)$ 。

输出格式

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

样例 #1

样例输入 #1

样例输出 #1

提示

时限 2 秒, 256M。蓝桥杯 2017 年第八届

思路

第一步：可以很容易想一个O（n^3）的暴力做法，枚举一遍区间的右端点，在枚举区间的左端点，在枚举左右端点之间所有的数，加起来判断是不是mod k为0。
第二步:进行第一步优化，利用前缀和数组把所有的前缀和预处理出来，这样就可以解决掉第三重循环，变成O（n^2）的复杂度。处理出前缀和数组s之后，求（l,r）区间的和公式为s[r]-s[l-1]；
第三步:由于数据范围是10^5，平方的复杂度肯定会超时，因此还需要在优化掉一重循环，这样可以使用空间换时间的方法。考虑，其实就是要在所有以r为右端点的区间里找到L使得
（s[r]-s[l-1]）%k==0 即在（0~r-1）区间内找到所有的L使得s[L]和s[r]的余数相同。

注意这个题目的数据范围：前缀和很可能会爆int，答案也可能会爆int，因此要用long long来存储！

代码

正解：前缀和+优化

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

using namespace std;
typedef long long LL;

const int N = 100010;
LL s[N],cnt[N];//s是前缀和数组，cnt[i]是记录余数为i的数的个数
int n,k;

int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%lld",&s[i]);
        s[i]+=s[i-1];//预处理前缀和
    }
    LL res=0;
    cnt[0]=1;
    for(int i=1;i<=n;i++){
        res+=cnt[s[i]%k];
        cnt[s[i]%k]++;
    }
    printf("%lld\n",res);
    return 0;
}