【蓝桥杯C++】k倍区间

最新推荐文章于 2024-04-05 18:22:15 发布

原创最新推荐文章于 2024-04-05 18:22:15 发布 · 593 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#蓝桥杯 #c++

该博客介绍了一种优化算法，通过前缀和和数学表达式转换将复杂度降低到O(N)。代码示例中展示了如何处理整数数组，并计算数组元素对k取模后的余数出现的次数，特别注意了整型溢出问题，采用long long类型避免。博客内容与编程技巧和算法优化相关。

本题为，巧用前缀和+数学表达式转换

优化为O(N)，

代码思路

#include <iostream>
#include <cstdio>
using namespace std;

const int N = 100010;
typedef long long ll;

ll s[N] ;
int cnt[N];

int n , k;

int main()
{
    scanf("%d%d" , &n , &k );
    for( int i = 1 ; i <= n ; i++ ) 
    {
        scanf("%d" , &s[i]); 
        s[i] += s[i-1];
    }
    
    ll res = 0;
    cnt[0] = 1;//cnt[0]中存的是s[]中等于0的数的个数 由于s[0] = 0 所以最初等于0的有1个数 所以cnt[0] = 1 
    for( int i = 1 ; i <= n ; i++ )
    {
        res += cnt[s[i] % k];//表示s[i]%k的余数出现的次数
        cnt[s[i] % k]++;
    }
    printf("%lld",res);
    return 0;
}

要注意的就是，int会爆栈，开long long