蓝桥杯 6. k倍区间

k倍区间

原题目链接

题目描述

给定一个长度为 N 的数列 A₁, A₂, ⋯, Aₙ,如果其中一段连续的子序列 Aᵢ, Aᵢ₊₁, ⋯, Aⱼ (i ≤ j) 之和是 K 的倍数,我们就称这个区间 [i, j]K 倍区间

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


输入描述

  • 第一行包含两个整数 NK (1 ≤ N, K ≤ 10⁵)。
  • 接下来 N 行,每行一个整数 Aᵢ (1 ≤ Aᵢ ≤ 10⁵),表示数列的每个元素。

输出描述

输出一个整数,表示 K 倍区间的数量。


输入样例

in 复制代码
5 2
1
2
3
4
5

输出样例

out 复制代码
6

c++代码

cpp 复制代码
#include<bits/stdc++.h>
#include<stdio.h>

using namespace std;

typedef long long ll;
ll N, K, ans = 0;
unordered_map<int, int> mp;
vector<ll> arr;

int main() {
    scanf("%lld %lld", &N, &K);
    arr = vector<ll>(N + 1, 0);
    for (ll i = 1; i <= N; i++) {
        scanf("%lld", &arr[i]);
        arr[i] += arr[i - 1];
    }
    if (K == 1) {
        printf("%lld", (N * (N - 1)) / 2 + N);
        return 0;
    }
    for (int i = 0; i <= N; i++) mp[arr[i] % K]++;
    for (auto it = mp.begin(); it != mp.end(); it++) ans += ((it->second * (it->second - 1)) / 2);
    printf("%lld", ans);
    return 0;
}//by wqs

算法解析

首先前缀和想得到吧

cpp 复制代码
for (ll i = 1; i <= N; i++) {
    scanf("%lld", &arr[i]);
    arr[i] += arr[i - 1];
}

现在有了前缀和数组

我们要找两个端点i, j,使得

cpp 复制代码
(arr[j] - arr[i]) % k == 0
假设arr[j] % k = c, arr[i] % k = d,
根据取模公式(arr[j] - arr[i]) % k = c - d。
也就是我们要让c - d = 0,也就是c == d。
现在我们的任务转换为求有多少组i,j满足arr[i] % k == arr[j] % k。
cpp 复制代码
for (int i = 0; i <= N; i++) mp[arr[i] % K]++;
for (auto it = mp.begin(); it != mp.end(); it++) ans += ((it->second * (it->second - 1)) / 2);//排列组合
printf("%lld", ans);
相关推荐
向阳@向远方3 分钟前
第二章 简单程序设计
开发语言·c++·算法
Mr_Xuhhh34 分钟前
信号与槽的总结
java·开发语言·数据库·c++·qt·系统架构
github_czy1 小时前
RRF (Reciprocal Rank Fusion) 排序算法详解
算法·排序算法
liulilittle1 小时前
VGW 虚拟网关用户手册 (PPP PRIVATE NETWORK 基础设施)
开发语言·网络·c++·网关·智能路由器·路由器·通信
许愿与你永世安宁1 小时前
力扣343 整数拆分
数据结构·算法·leetcode
爱coding的橙子1 小时前
每日算法刷题Day42 7.5:leetcode前缀和3道题,用时2h
算法·leetcode·职场和发展
ruanjiananquan991 小时前
c,c++语言的栈内存、堆内存及任意读写内存
java·c语言·c++
满分观察网友z2 小时前
从一次手滑,我洞悉了用户输入的所有可能性(3330. 找到初始输入字符串 I)
算法
持梦远方2 小时前
C 语言基础入门:基本数据类型与运算符详解
c语言·开发语言·c++
YuTaoShao2 小时前
【LeetCode 热题 100】73. 矩阵置零——(解法二)空间复杂度 O(1)
java·算法·leetcode·矩阵