【蓝桥杯JavaB组真题详解】k倍区间（2017）

最新推荐文章于 2023-01-15 19:20:09 发布

疼疼蛇

最新推荐文章于 2023-01-15 19:20:09 发布

阅读量722

点赞数 3

分类专栏：蓝桥杯JavaB组真题详解文章标签： java 算法数据结构 c++ 蓝桥杯

本文链接：https://blog.csdn.net/m0_46226318/article/details/115406187

版权

蓝桥杯JavaB组真题详解专栏收录该内容

63 篇文章 39 订阅

订阅专栏

这是一个关于算法的博客，主要讨论如何解决寻找一个数列中K倍区间的数量问题。给定一个数列和整数K，通过计算数列的前缀和并利用模K的余数，可以有效地找出所有连续子序列的和为K倍数的区间。程序使用了前缀和技巧和动态规划思想，输出了满足条件的区间总数。

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

题目描述

k倍区间
给定一个长度为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

资源约定：
峰值内存消耗（含虚拟机） < 256M
CPU消耗 < 2000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入…” 的多余内容。

所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。
不要使用package语句。不要使用jdk1.7及以上版本的特性。
主类的名字必须是：Main，否则按无效代码处理。

题目解析

利用前缀和模除k记录于数组，
再根据两个余数相同的可以随意组合算出结果
参考代码


import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int k = sc.nextInt();
		int[] arr = new int[n];
		long[] v = new long[200000];
		long sum = 0;
		long count=0;
		for (int i = 0; i < n; i++) {
			arr[i] = sc.nextInt();
			sum += arr[i];
			v[(int) (sum % k)]++;
		}
		sc.close();
		count += v[0];
		for (int i = 0; i < k; i++) {
			count += (v[i]*(v[i]-1))/2;
		}
		System.out.println(count);
	}

}