BOJ 22967 - 구름다리

BOJ 22967 - 구름다리

완전 그래프의 간선 수는 $\dfrac{N(N-1)}{2}$이기 때문에 $2(N-1)$ 이 이것 이상이지 않으면 $R > 1$ 이다.

그런데 $0$번 건물로만 모두 연결해버리면 항상 $R=2$ 를 만들 수 있기 때문에 이 두 경우를 처리해주면 된다.

void solve() {
   int n;
   cin >> n;
   vvi g(n, vi(n));
   for (int i = 0; i < n - 1; i++) {
      int u, v;
      cin >> u >> v;
      u--, v--;
      g[u][v] = g[v][u] = 1;
   }
   vector<pi> ans;
   int R = 0;
   for (int i = 0; i < n; i++) g[i][i] = 1;
   if (n - 1 + n - 1 >= n * (n - 1) / 2) {
      R = 1;
      for (int i = 0; i < n; i++) {
         for (int j = i + 1; j < n; j++) {
            if (!g[i][j]) {
               ans.pb({i + 1, j + 1});
               g[i][j] = g[j][i] = 1;
            }
         }
      }
   } else {
      R = 2;
      for (int i = 0; i < n; i++) {
         if (!g[i][0]) {
            g[i][0] = g[0][i] = 1;
            ans.pb({i + 1, 1});
         }
      }
   }
   cout << sz(ans) << endl << R << endl;
   for (auto &[u, v]: ans) cout << u << ' ' << v << endl;
}