B - 抽屉 POJ - 2356 (容斥原理)

The input contains N natural (i.e. positive integer) numbers ( N <= 10000 ). Each of that numbers is not greater than 15000. This numbers are not necessarily different (so it may happen that two or more of them will be equal). Your task is to choose a few of given numbers ( 1 <= few <= N ) so that the sum of chosen numbers is multiple for N (i.e. N * k = (sum of chosen numbers) for some natural number k).

Input

The first line of the input contains the single number N. Each of next N lines contains one number from the given set.

Output

In case your program decides that the target set of numbers can not be found it should print to the output the single number 0. Otherwise it should print the number of the chosen numbers in the first line followed by the chosen numbers themselves (on a separate line each) in arbitrary order. 

If there are more than one set of numbers with required properties you should print to the output only one (preferably your favorite) of them.

Sample Input

5
1
2
3
4
1

Sample Output

2
2
3

 

题意：

输入n

输入n个数

判断这n个数中是不是有几个数字之和是n的倍数

思路：

n个数余数分别为 1 ~ n-1 ,相当于有n-1个抽屉，n个物品

分别计算a[1] + a[2] + …… + a[k] 的和然后取余如果为零则直接输出前k个数，否则寻找余数相同的两个数，假设为i， j （i < j),则a[i+1] + . . . . + a[j] 的和一定能被n整除（原理还没想清楚）

 

AC代码

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
int a[10005];
int mod[10005];
int mark[10005];

int main()
{
    int n;
    bool flag = false;
    cin >> n;
    memset(mod, 0, sizeof(mod));
    memset(mark, 0, sizeof(mark));
    for(int i = 1; i <= n; i++)
    {
        cin >> a[i];
        mod[i] = (mod[i-1] + a[i]) % n;
    }

    for(int i = 1; i <= n; i++)
    {
        if(mod[i] == 0)
        {
            flag = true;
            cout << i << endl;
            for(int j = 1; j <= i; j++)
                cout << a[j] << endl;
            break;
        }
    }

    if(!flag)
    {
        for(int i = 1; i <= n; i++)
        {
            if(mark[mod[i]] == 0)
                mark[mod[i]] = i;
            else
            {
                cout << i -mark[mod[i]] << endl;
                for(int j = mark[mod[i]]+1; j <= i; j++)
                    cout << a[j] << endl;

                break;
            }
        }
    }

    return 0;
}