题目描述
对于小宝来说,如果一个数组的总和能够整除他的幸运数字k,就是他的幸运数组,而其他数组小宝都很讨厌。现在有一个长度为n的数组,小宝想知道这个数组的子数组中,最长的幸运子数组有多长。
对于子数组的定义,如果可以通过从开头和从结束分别删除若干个(可以为零或全部,前后删除个数不必相同)元素来从数组b获得数组a,则称数组a是数组b的子数组。(子数组包含原数组,但不包含空串)
输入描述:
多组输入。第一行包含一个整数 T ( 1 ≤ T ≤ 10 ) T(1≤T≤10) T(1≤T≤10),表示有T组测试数据。
每组测试数据包含两行,第一行包含两个整数n和 k ( 1 ≤ n ≤ 1 0 5 , 1 ≤ k ≤ 1 0 5 ) k(1≤n≤10^5,1≤k≤10^5) k(1≤n≤105,1≤k≤105),分别表示数组长度和小宝的幸运数字。第二行包含n个空格分隔的整数 a 1 , a 2 , . , a n ( 0 ≤ a i ≤ 1 0 9 ) a1,a2,.,an(0≤ai≤10^9) a1,a2,.,an(0≤ai≤109),为数组的元素。
输出描述:
对于每组数据,输出和能被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
AC的C++代码
#include <iostream>
#include <memory.h>
using namespace std;
int main() {
int a,T,k,n;
int sum[100010] = {0};
cin >> T;
int pos[100010] = {-1};
while (T-- > 0) {
memset(pos, -1, sizeof(pos));
pos[0] = 0;
int Max = -1;
cin >> n >> k;
for (int i = 1; i <= n; ++i) {
cin >> a;
sum[i] = sum[i - 1] + a;
sum[i] %= k;
if (pos[sum[i]] == -1) {
pos[sum[i]] = i;
} else {
Max = max(Max, (int) (i - pos[sum[i]]));
}
}
cout << Max << endl;
}
}