結果
| 問題 | 
                            No.2638 Initial fare
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2024-02-19 22:29:31 | 
| 言語 | C++17  (gcc 13.3.0 + boost 1.87.0)  | 
                    
| 結果 | 
                             
                                TLE
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 853 bytes | 
| コンパイル時間 | 920 ms | 
| コンパイル使用メモリ | 81,344 KB | 
| 最終ジャッジ日時 | 2025-02-19 17:16:16 | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge5 / judge2 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 3 | 
| other | AC * 4 TLE * 1 -- * 20 | 
ソースコード
#include <iostream>
#include <vector>
#include <deque>
using namespace std;
int main() {
    int N;
    cin >> N;
    vector<vector<int>> edge(N);
    for (int i = 0; i < N - 1; ++i) {
        int u, v;
        cin >> u >> v;
        u--;
        v--;
        edge[u].push_back(v);
        edge[v].push_back(u);
    }
    int ans = 0;
    deque<int> nxt;
    vector<bool> used(N, false);
    used[0] = true;
    nxt.push_back(0);
    while (!nxt.empty()) {
        int pos = nxt.back();
        nxt.pop_back();
        for (int to : edge[pos]) {
            ans++;
            for (int nt : edge[to]) {
                if (nt != pos) ans += edge[nt].size();
            }
            if (!used[to]) {
                nxt.push_back(to);
                used[to] = true;
            }
        }
    }
    cout << ans / 2 << endl;
    return 0;
}