結果
問題 |
No.1103 Directed Length Sum
|
ユーザー |
![]() |
提出日時 | 2020-07-27 19:06:19 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 636 ms / 3,000 ms |
コード長 | 2,063 bytes |
コンパイル時間 | 2,149 ms |
コンパイル使用メモリ | 198,412 KB |
最終ジャッジ日時 | 2025-01-12 06:57:33 |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define int ll #define ll long long #define I32_MAX 2147483647 #define I64_MAX 9223372036854775807LL #define I64_MAX2 1223372036854775807LL #define INF I64_MAX2 #define MOD 1000000007 // #define MOD 998244353 #define MEM_SIZE 100010 #define DEBUG_OUT true #define ALL(x) (x).begin(), (x).end() template<typename T> void DEBUG(T e){if(DEBUG_OUT == false)return; std::cout << e <<" ";} template<typename T> void DEBUG(const std::vector<T>& v){if(DEBUG_OUT == false)return;for(const auto& e : v){std::cout<< e << " "; } std::cout << std::endl;} template<typename T> void DEBUG(const std::vector<std::vector<T> >& vv){if(DEBUG_OUT == false)return;for(const auto& v : vv){ DEBUG(v); } } template<class T,class... Ts> void DEBUG(T d, Ts... e){if(DEBUG_OUT == false)return;DEBUG(d);DEBUG(e...);} template <class T> void corner(bool flg, T hoge) {if (flg) {cout << hoge << endl; abort();}} template< typename T1, typename T2 > inline bool chmax(T1 &a, T2 b) { return a < b && (a = b, true); } template< typename T1, typename T2 > inline bool chmin(T1 &a, T2 b) { return a > b && (a = b, true); } void DFS(int V,int P, int D,vector<vector<int> > &GRAPH ,vector<int> & data) { if(P != -1) { data[V] = data[P] + D; data[V] %= MOD; } for (auto &&nv : GRAPH[V]) { DFS(nv,V,D+1,GRAPH,data); } return; } void solve(void) { int N,M; cin>>N; M = N -1; vector<vector<int> > GRAPH (N); vector<int> P(N,0); for (int i = 0; i < M; i++) { int A,B; cin>>A>>B; A--;B--; GRAPH[A].push_back(B); P[B]++; } int root = -1; for (int i = 0; i < N; i++) { if(P[i] == 0)root = i; } vector<int> data (N,0); DFS(root,-1,0,GRAPH,data); int res = 0; for (auto &&x : data) { res += x; res %= MOD; } // DEBUG(data); cout<<res<<endl; return; } int32_t main(int32_t argc, const char *argv[]) { std::ios::sync_with_stdio(false); std::cin.tie(0); std::cout << std::fixed; std::cout << std::setprecision(11); solve(); return 0; }