蓝桥杯 k倍区间

最新推荐文章于 2025-02-19 22:25:31 发布

原创最新推荐文章于 2025-02-19 22:25:31 发布 · 84 阅读

0 ·

CC 4.0 BY-SA版权

这是一个Java程序，用于计算一个整数数组中，长度为K的不重复子区间（子数组）的数量，使得子区间的元素之和能被K整除。程序首先读取输入的数组长度n和整数k，然后计算数组的前缀和，并使用哈希映射记录每个余数值出现的次数。最后，通过计算每个余数值不重复子区间的数量，得出总答案。

package 第八届;

import java.util.Scanner;

//2021年3月31日上午11:48:49
//writer:apple

import java.util.*;
import java.io.*;
public class K倍区间 {

	static int n;
	static int k;
	static long sum[];
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner s=new  Scanner(System.in);
		n=s.nextInt();k=s.nextInt();
		sum=new long[n+1];
		for(int i=1;i<=n;i++)
		{
			sum[i]=s.nextInt();
			sum[i]+=sum[i-1];
		}
		long ans=0;
		int cnt[]=new int[k];
		for(int i=1;i<=n;i++)
		{
			sum[i]%=k;
			cnt[(int) sum[i]]++;
		}
		ArrayList<Integer> cntc=new ArrayList<>();
		for(int b:cnt) 
			{
			if(b>1)
			cntc.add(b);
			}
		ans+=cnt[0];
		for(long t:cntc) 
		{
			ans+=(t*(t-1)/2);
		}
		
		System.out.println(ans);
	}
}
//O(N^2)
//long ans=0;
//for(int i=1;i<=n;i++)
//{
//	
//	for(int j=i;j<=n;j++)//[i,j)  j-i   i+1...j
//	{
//		long ss=sum[j]-sum[i-1];
//		if(ss%k==0) ans++;
//	}
//}