小宝的幸运数组

小宝的幸运数组

题目链接

题目描述

对于小宝来说，如果一个数组的总和能够整除他的幸运数字k，就是他的幸运数组，而其他数组小宝都很讨厌。现在有一个长度为n的数组，小宝想知道这个数组的子数组中，最长的幸运子数组有多长。
对于子数组的定义，如果可以通过从开头和从结束分别删除若干个（可以为零或全部，前后删除个数不必相同）元素来从数组b获得数组a，则称数组a是数组b的子数组。（子数组包含原数组，但不包含空串）

输入格式

多组输入。第一行包含一个整数T(1≤T≤10)，表示有T组测试数据。
每组测试数据包含两行，第一行包含两个整数n和k(1≤n≤10^5,1≤k≤105)，分别表示数组长度和小宝的幸运数字。第二行包含n个空格分隔的整数a1,a2,.,an(0≤ai≤10^9)，为数组的元素。

输出格式

对于每组数据，输出和能被k整除的最长子数组的长度。如果没有这样的子数组，则输出−1。

示例

输入

4
3 3
1 2 3
3 5
1 2 3
3 7
1 2 3
1 6
5

输出

3
2
-1
-1

解题思路

前言

首先,要掌握前缀和的概念,不明白这个概念的可以去自己百度搜索相关资料。在掌握前缀和的情况下需要掌握(a-b)%k==a%k-b%k的数学基本知识。在此基础上，我们只需要通过贪心算法求解即可。

详解

在掌握前言的基础上，我们在对前缀和求模的时候，将模的下标保存到哈希表(数组)中，一旦存在之后其他前缀和的模在哈希表中存在，则证明两个下标之间的数组内容和能够使得被k整除(查看前面的数学知识)。

代码

import java.io.*;
import java.util.*;
public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
		StreamTokenizer st=new StreamTokenizer(br);
		PrintWriter pr=new PrintWriter(new BufferedWriter(new OutputStreamWriter(System.out)));
		st.nextToken();
		int m=(int)st.nval;
		while(m>0) {
			m--;
			st.nextToken();
			int n=(int)st.nval;
			st.nextToken();
			int k=(int)st.nval;
			int[] preSumMods=new int[n+1];
			int[] greed=new int[k+1];
			Arrays.fill(greed, -1);
			greed[0]=0;
			int res=-1;
			//录入数据 并进行处理
			for(int i=1;i<=n;i++) {
				st.nextToken();
				preSumMods[i]=(preSumMods[i-1]+(int)st.nval)%k;
				if(greed[preSumMods[i]]!=-1) res=Math.max(res, i-greed[preSumMods[i]]);
				else greed[preSumMods[i]]=i;
			}
			pr.println(res);
		}
		pr.flush();
		return;
	}
}