[51Nod1103]N的倍数

传送门51Nod1103

Description

一个长度为N的数组A，从A中选出若干个数，使得这些数的和是N的倍数。
例如：N = 8，数组A包括：2 5 6 3 18 7 11 19，可以选2 6，因为2 + 6 = 8，是8的倍数。
如果有多组合法方案, 输出任意一组即可.

Input

第1行：1个数N，N为数组的长度，同时也是要求的倍数。(2 <= N <= 50000)
第2 - N + 1行：数组A的元素。(0 < A[i] <= 10^9)

Output

如果没有符合条件的组合，输出No Solution。
第1行：1个数S表示你所选择的数的数量。
第2 - S + 1行：每行1个数，对应你所选择的数。

Sample Input

8
2
5
6
3
18
7
11
19

Sample Output

2
2
6

Solution

• 其实是个抽屉原理…
• 求出s[i]表示前i位和模n的结果
• 一共求出n个取模结果，如果各不相同，必有一个s[i]为0，答案即为1至i。如果有s[i]=s[j]，答案为i到j
• 另一个版本n有1000000+，读入优化了解一下…

---

Code

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
int n,ans,l,r,s[1000007],a[1000007];
int p[1000007];
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (!(ch>='0'&&ch<='9')){if (ch='-') f=-1; ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int main(){
    n=read();
    for (int i=1;i<=n;i++){
        a[i]=read();
        s[i]=(s[i-1]+a[i])%n;
    }
    for (int i=1;i<=n;i++){
        if (!s[i]){l=1; r=i; break;}
        if (p[s[i]]){l=p[s[i]]+1; r=i; break;}
        p[s[i]]=i;
    }
    ans=r-l+1;
    printf("%d\n",ans);
    for (int i=l;i<=r;i++) printf("%d\n",a[i]);
    return 0;
}