結果
問題 |
No.778 クリスマスツリー
|
ユーザー |
![]() |
提出日時 | 2019-04-08 21:52:47 |
言語 | C++11(廃止可能性あり) (gcc 13.3.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 953 bytes |
コンパイル時間 | 1,454 ms |
コンパイル使用メモリ | 166,780 KB |
実行使用メモリ | 40,960 KB |
最終ジャッジ日時 | 2024-07-01 22:54:57 |
合計ジャッジ時間 | 5,550 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 4 WA * 8 |
ソースコード
#include <bits/stdc++.h> #define _overload3(_1, _2, _3, name, ...) name #define _rep(i, n) repi(i, 0, n) #define repi(i, a, b) for (int i = (int)(a); i < (int)(b); ++i) #define rep(...) _overload3(__VA_ARGS__, repi, _rep, )(__VA_ARGS__) #define ll long long #define lld long double #ifdef DEBUG #define line() cout << "[" << __LINE__ << "] "; #define dump(i) cout << #i ": " << i << " "; #define dumpl(i) cout << #i ": " << i << endl; #else #define line(i) #define dump(i) #define dumpl(i) #endif using namespace std; ll calc(map<int, vector<int>> &mp, int index, int level) { if (mp[index].empty()) return level; ll ret = 0; for (int i : mp[index]) { ret += calc(mp, i, level + 1); } return ret; } int main() { int n; cin >> n; map<int, vector<int>> mp; rep(i, n - 1) { int t; cin >> t; mp[t].push_back(i + 1); } cout << calc(mp, 0, 0) << endl; return 0; }