結果

問題 No.1928 Make a Binary Tree
ユーザー 👑 emthrmemthrm
提出日時 2022-05-06 22:24:06
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,087 bytes
コンパイル時間 2,594 ms
コンパイル使用メモリ 209,016 KB
実行使用メモリ 45,304 KB
最終ジャッジ日時 2023-09-20 03:21:56
合計ジャッジ時間 9,709 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 2 ms
4,380 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 1 ms
4,376 KB
testcase_06 WA -
testcase_07 AC 1 ms
4,380 KB
testcase_08 AC 1 ms
4,380 KB
testcase_09 AC 2 ms
4,376 KB
testcase_10 AC 1 ms
4,376 KB
testcase_11 AC 1 ms
4,380 KB
testcase_12 AC 1 ms
4,376 KB
testcase_13 WA -
testcase_14 AC 2 ms
4,376 KB
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 AC 66 ms
20,316 KB
testcase_36 AC 75 ms
33,984 KB
testcase_37 AC 114 ms
28,536 KB
testcase_38 AC 1 ms
4,376 KB
testcase_39 AC 131 ms
35,148 KB
testcase_40 AC 128 ms
35,228 KB
testcase_41 AC 126 ms
35,144 KB
testcase_42 AC 124 ms
35,300 KB
testcase_43 AC 121 ms
35,164 KB
testcase_44 AC 123 ms
35,248 KB
testcase_45 AC 128 ms
35,420 KB
testcase_46 AC 126 ms
35,092 KB
testcase_47 AC 125 ms
35,104 KB
testcase_48 AC 127 ms
34,944 KB
testcase_49 AC 87 ms
45,304 KB
testcase_50 AC 213 ms
30,060 KB
testcase_51 AC 211 ms
30,036 KB
testcase_52 AC 208 ms
30,044 KB
testcase_53 AC 215 ms
30,100 KB
testcase_54 AC 220 ms
29,960 KB
testcase_55 AC 88 ms
36,816 KB
testcase_56 WA -
testcase_57 WA -
testcase_58 WA -
testcase_59 AC 90 ms
22,560 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
using namespace std;
#define FOR(i,m,n) for(int i=(m);i<(n);++i)
#define REP(i,n) FOR(i,0,n)
#define ALL(v) (v).begin(),(v).end()
using ll = long long;
constexpr int INF = 0x3f3f3f3f;
constexpr long long LINF = 0x3f3f3f3f3f3f3f3fLL;
constexpr double EPS = 1e-8;
constexpr int MOD = 1000000007;
// constexpr int MOD = 998244353;
constexpr int DY4[]{1, 0, -1, 0}, DX4[]{0, -1, 0, 1};
constexpr int DY8[]{1, 1, 0, -1, -1, -1, 0, 1};
constexpr int DX8[]{0, -1, -1, -1, 0, 1, 1, 1};
template <typename T, typename U>
inline bool chmax(T& a, U b) { return a < b ? (a = b, true) : false; }
template <typename T, typename U>
inline bool chmin(T& a, U b) { return a > b ? (a = b, true) : false; }
struct IOSetup {
  IOSetup() {
    std::cin.tie(nullptr);
    std::ios_base::sync_with_stdio(false);
    std::cout << fixed << setprecision(20);
  }
} iosetup;

int main() {
  int n; cin >> n;
  vector<vector<int>> graph(n);
  REP(_, n - 1) {
    int x, y; cin >> x >> y; --x; --y;
    graph[x].emplace_back(y);
    graph[y].emplace_back(x);
  }
  vector<priority_queue<int>> que(n);
  auto f = [&](auto&& f, int par, int ver) -> int {
    if (par != -1) graph[ver].erase(find(ALL(graph[ver]), par));
    if (graph[ver].empty()) return 1;
    if (graph[ver].size() == 1) {
      int res = f(f, ver, graph[ver].front()) + 1;
      que[ver].swap(que[graph[ver].front()]);
      if (!que[ver].empty()) {
        res += que[ver].top(); que[ver].pop();
      }
      return res;
    }
    int top1 = 0, top2 = 0;
    for (int e : graph[ver]) {
      const int child = f(f, ver, e);
      if (child > top1) {
        que[ver].emplace(top2);
        top2 = top1;
        top1 = child;
      } else if (child > top2) {
        que[ver].emplace(top2);
        top2 = child;
      } else {
        que[ver].emplace(child);
      }
      if (que[ver].size() < que[e].size()) swap(que[ver], que[e]);
      for (; !que[e].empty(); que[e].pop()) que[ver].emplace(que[e].top());
    }
    return top1 + top2 + 1;
  };
  cout << f(f, -1, 0) << '\n';
  return 0;
}
0