結果
問題 | No.2638 Initial fare |
ユーザー | MMRZ |
提出日時 | 2024-02-19 21:52:03 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 90 ms / 2,000 ms |
コード長 | 2,571 bytes |
コンパイル時間 | 2,725 ms |
コンパイル使用メモリ | 251,164 KB |
実行使用メモリ | 16,488 KB |
最終ジャッジ日時 | 2024-09-29 01:50:44 |
合計ジャッジ時間 | 5,597 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 |
ソースコード
# include <bits/stdc++.h> using namespace std; using ll = long long; using ull = unsigned long long; const double pi = acos(-1); template<class T>constexpr T inf() { return ::std::numeric_limits<T>::max(); } template<class T>constexpr T hinf() { return inf<T>() / 2; } template <typename T_char>T_char TL(T_char cX) { return tolower(cX); } template <typename T_char>T_char TU(T_char cX) { return toupper(cX); } template<class T> bool chmin(T& a,T b) { if(a > b){a = b; return true;} return false; } template<class T> bool chmax(T& a,T b) { if(a < b){a = b; return true;} return false; } template<class T> bool is_sqare(T a) { if(floor(sqrt(a)) * floor(sqrt(a)) == a){ return true; }return false; } int popcnt(unsigned long long n) { int cnt = 0; for (int i = 0; i < 64; i++)if ((n >> i) & 1)cnt++; return cnt; } int d_sum(ll n) { int ret = 0; while (n > 0) { ret += n % 10; n /= 10; }return ret; } int d_cnt(ll n) { int ret = 0; while (n > 0) { ret++; n /= 10; }return ret; } ll gcd(ll a, ll b) { if (b == 0)return a; return gcd(b, a%b); }; ll lcm(ll a, ll b) { ll g = gcd(a, b); return a / g*b; }; template<class T> using dijk = priority_queue<T, vector<T>, greater<T>>; # define all(qpqpq) (qpqpq).begin(),(qpqpq).end() # define UNIQUE(wpwpw) (wpwpw).erase(unique(all((wpwpw))),(wpwpw).end()) # define LOWER(epepe) transform(all((epepe)),(epepe).begin(),TL<char>) # define UPPER(rprpr) transform(all((rprpr)),(rprpr).begin(),TU<char>) # define rep(i,upupu) for(ll i = 0, i##_len = (upupu);(i) < (i##_len);(i)++) # define reps(i,opopo) for(ll i = 1, i##_len = (opopo);(i) <= (i##_len);(i)++) # define len(x) ((ll)(x).size()) # define bit(n) (1LL << (n)) # define pb push_back # define exists(c, e) ((c).find(e) != (c).end()) #ifdef LOCAL # include "_debug_print.hpp" # define debug(...) debug_print::multi_print(#__VA_ARGS__, __VA_ARGS__) #else # define debug(...) (static_cast<void>(0)) #endif struct INIT{ INIT(){ std::ios::sync_with_stdio(false); std::cin.tie(0); cout << fixed << setprecision(20); } }INIT; void solve(){ int n; cin >> n; vector<vector<int>> g(n); vector<int> a(n-1), b(n-1); rep(i, n-1){ cin >> a[i] >> b[i]; a[i]--, b[i]--; g[a[i]].pb(b[i]); g[b[i]].pb(a[i]); } ll ans3 = 0, ans2 = 0, ans1 = 0; rep(i, n-1){ ans3 += (len(g[a[i]])-1)*(len(g[b[i]])-1); ans2 += (len(g[a[i]])-1); ans2 += (len(g[b[i]])-1); ans1++; } cout << ans2/2 + ans1 + ans3 << endl; } int main(){ int t = 1; //cin >> t; while(t--)solve(); }