51Nod1103 N的倍数 前缀和+抽屉定理

题目链接:这里写链接内容
一个长度为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个数,对应你所选择的数。
Input示例
8
2
5
6
3
18
7
11
19
Output示例
2
2
6

这个题没说要最小的,只要输出n的倍数就行,不知道oj怎么判定的哈哈,从数组中选出若干数为n的倍数,可以先求出前缀和,有n个前缀和,但是余数只有1到n-1,所以一共有n-1种余数,根据抽屉定理,肯定有两个以上是同样的余数,只要找到这两个前缀和,[1,i]和[1,j],那么就可以知道[i+1,j]区间里的元素和符合题目要求。

#include<iostream>
#include<cstdio>

using namespace std;
typedef long long ll;
const int maxn=50000+10;
int sum;
int a[maxn];
int vis[maxn];
int main() {
    int n;
    scanf("%d",&n);
    for(int i=0; i<n; i++) {
        scanf("%d",&a[i]);
    }
    for(int i=0; i<n; i++) {
        sum=(sum+a[i])%n;
        if(!sum) {
            printf("%d\n",i+1);
            for(int j=0; j<=i; j++) {
                printf("%d\n",a[j]);
            }
            break;
        } else if(vis[sum]) {
            printf("%d\n",i-vis[sum]);
            for(int j=vis[sum]+1; j<=i; j++) {
                printf("%d\n",a[j]);
            }
            break;
        }
        vis[sum]=i;
    }

}

猜你喜欢

转载自blog.csdn.net/qq_37774171/article/details/82048894