ferinの競プロ帳

競プロについてのメモ

RUPC day1 A: 鳩ノ巣原理

問題ページ
Aizu Online Judge Arena

概要

N個の自然数a[i]が与えられる
abs(a[i]-a[j]) = 0 (mod n-1) を満たすような組(i!=j)を一つ示せ

解法

N <= 1000 でO(N^2)が間に合うので全てのペアについて探索すればよい

ソースコード

#include <bits/stdc++.h>

using namespace std;
using ll = long long;
#define int ll
using VI = vector<int>;

#define FOR(i, a, n) for (ll i = (ll)a; i < (ll)n; ++i)
#define REP(i, n) FOR(i, 0, n)

signed main(void)
{
  cin.tie(0);
  ios::sync_with_stdio(false);

  int n;
  cin >> n;
  VI a(n);
  REP(i, n) cin >> a[i];

  REP(i, n) FOR(j, i+1, n) {
    if(abs(a[i]-a[j])%(n-1) == 0) {
      cout << a[i] << " " << a[j] << endl;
      return 0;
    }
  }

  return 0;
}