第八届“图灵杯”NEUQ-ACM程序设计竞赛个人赛——B题 小宝的幸运数组

题目描述

对于小宝来说，如果一个数组的总和能够整除他的幸运数字k，就是他的幸运数组，而其他数组小宝都很讨厌。现在有一个长度为n的数组，小宝想知道这个数组的子数组中，最长的幸运子数组有多长。

对于子数组的定义，如果可以通过从开头和从结束分别删除若干个（可以为零或全部，前后删除个数不必相同）元素来从数组b获得数组a，则称数组a是数组b的子数组。（子数组包含原数组，但不包含空串）

输入描述:

多组输入。第一行包含一个整数$T(1\le T\le 10)$，表示有T组测试数据。

每组测试数据包含两行，第一行包含两个整数n和$k(1\le n\le 10^5,1\le k\le 10^5)$，分别表示数组长度和小宝的幸运数字。第二行包含n个空格分隔的整数$a1,a2,.,an(0\le ai\le 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

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;
    }
}