oj.uz

Submission #316956

# Submission time Handle Problem Language Result Execution time Memory
316956 2020-10-28T17:24:36 Z AlexLuchianov Countries (BOI06_countries) C++14
100 / 100
 62 ms 512 KB 
#include <iostream>
#include <vector>
#include <cassert>
#include <algorithm>
#include <cmath>

using ll = long long;
#define MIN(a, b) (((a) < (b)) ? (a) : (b))
#define MAX(a, b) (((a) < (b)) ? (b) : (a))
int const nmax = 1000;

struct Point{
  int x;
  int y;
  int s;
} v[1 + nmax];

struct Number{
  int x;
  int y;
  Number(int x_ = 0, int y_ = 1) {
    int d = std::__gcd(x_, y_);

    x = x_ / d;
    y = y_ / d;
  }
  bool operator < (Number a) {
    return 1LL * x * a.y < 1LL * a.x * y;
  }
  bool operator == (Number a) {
    return 1LL * x * a.y == 1LL * a.x * y;
  }
};

int dist(Point a, Point b) {
  return (a.x - b.x) * (a.x - b.x) + (a.y - b.y) * (a.y - b.y);
}

Number exert(Point a, Point b) {
  return {a.s, dist(a, b)};
}

int leader[1 + nmax], _count[1 + nmax];
Number emass[1 + nmax];

int dfs(int node) {
  if(leader[node] != node)
    leader[node] = dfs(leader[node]);
  return leader[node];
}

int main() {
  std::ios::sync_with_stdio(0);
  std::cin.tie(0);

  int n;
  std::cin >> n;
  for(int i = 1;i <= n; i++)
    std::cin >> v[i].x >> v[i].y >> v[i].s;
  
  for(int i = 1;i <= n; i++)
    for(int j = 1;j <= n; j++)
      if(i != j) {
        Number newmass = exert(v[i], v[j]);
        if(emass[j] < newmass) {
          emass[j] = newmass;
          leader[j] = i;
          _count[j] = 1;
        } else if(emass[j] == newmass)
          _count[j]++;
      }
  
  for(int i = 1;i <= n; i++)
    if(emass[i] < Number(v[i].s, 1) || emass[i] == Number(v[i].s, 1) || 1 < _count[i])
      leader[i] = i;

  for(int i = 1;i <= n; i++)
    if(emass[i] < Number(v[i].s, 1) || emass[i] == Number(v[i].s, 1))
      std::cout << "K\n";
    else if(_count[i] == 1)
      std::cout << dfs(i) << '\n';
    else
      std::cout << "D\n";
  return 0;
}

Subtask #1
100.0 / 100.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 384 KB Output is correct
2 Correct 2 ms 384 KB Output is correct
3 Correct 4 ms 384 KB Output is correct
4 Correct 6 ms 384 KB Output is correct
5 Correct 15 ms 384 KB Output is correct
6 Correct 13 ms 384 KB Output is correct
7 Correct 28 ms 384 KB Output is correct
8 Correct 37 ms 512 KB Output is correct
9 Correct 46 ms 384 KB Output is correct
10 Correct 57 ms 384 KB Output is correct
11 Correct 1 ms 384 KB Output is correct
12 Correct 2 ms 384 KB Output is correct
13 Correct 6 ms 384 KB Output is correct
14 Correct 6 ms 384 KB Output is correct
15 Correct 16 ms 384 KB Output is correct
16 Correct 23 ms 384 KB Output is correct
17 Correct 31 ms 384 KB Output is correct
18 Correct 21 ms 512 KB Output is correct
19 Correct 19 ms 384 KB Output is correct
20 Correct 62 ms 384 KB Output is correct