Codeforces 876B Divisiblity of Differences：數學【任意兩數之差為k的倍數】

題目連結：http://codeforces.com/contest/876/problem/B

題意：

　　給你n個數a[i]，讓你找出一個大小為k的集合，使得集合中的數兩兩之差為m的倍數。

　　若有多解，輸出任意一個集合即可。

 

題解：

　　若一個集合中的數，兩兩之差為m的倍數，則他們 mod m 的值均相等。

　　所以O(N)掃一遍，對於每個數a：vector v[a%m].push_back(a)

　　一旦有一個集合大小為k，則輸出。

 

AC Code:

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <vector>
#define MAX_M 100005

using namespace std;

int n,k,m;
vector<int> v[MAX_M];

int main()
{
    cin>>n>>k>>m;
    int a;
    for(int i=0;i<n;i++)
    {
        cin>>a;
        v[a%m].push_back(a);
        if(v[a%m].size()==k)
        {
            cout<<"Yes"<<endl;
            for(int j=0;j<v[a%m].size();j++)
            {
                cout<<v[a%m][j]<<" ";
            }
            cout<<endl;
            return 0;
        }
    }
    cout<<"No"<<endl;
}