結果
問題 | No.1103 Directed Length Sum |
ユーザー | suzuken_w |
提出日時 | 2020-07-03 23:07:32 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,776 bytes |
コンパイル時間 | 1,962 ms |
コンパイル使用メモリ | 178,548 KB |
実行使用メモリ | 148,876 KB |
最終ジャッジ日時 | 2024-09-17 04:47:00 |
合計ジャッジ時間 | 7,702 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 16 ms
37,132 KB |
testcase_01 | AC | 17 ms
35,864 KB |
testcase_02 | WA | - |
testcase_03 | AC | 169 ms
43,232 KB |
testcase_04 | AC | 295 ms
46,632 KB |
testcase_05 | AC | 536 ms
55,112 KB |
testcase_06 | AC | 183 ms
44,020 KB |
testcase_07 | AC | 43 ms
38,656 KB |
testcase_08 | AC | 60 ms
39,476 KB |
testcase_09 | AC | 33 ms
37,916 KB |
testcase_10 | AC | 86 ms
40,340 KB |
testcase_11 | AC | 330 ms
47,500 KB |
testcase_12 | AC | 180 ms
42,864 KB |
testcase_13 | AC | 88 ms
39,312 KB |
testcase_14 | AC | 29 ms
36,868 KB |
testcase_15 | AC | 126 ms
40,680 KB |
testcase_16 | AC | 378 ms
49,244 KB |
testcase_17 | AC | 384 ms
49,760 KB |
testcase_18 | AC | 81 ms
38,884 KB |
testcase_19 | AC | 341 ms
48,000 KB |
testcase_20 | AC | 36 ms
37,596 KB |
testcase_21 | AC | 54 ms
37,536 KB |
testcase_22 | AC | 262 ms
47,204 KB |
testcase_23 | AC | 150 ms
41,156 KB |
ソースコード
#pragma GCC optimize("O3") #include<bits/stdc++.h> using namespace std; using ll=long long; using P=pair<ll,ll>; template<class T> using V=vector<T>; #define fi first #define se second #define all(v) (v).begin(),(v).end() const ll inf=(1e18); const ll mod=998244353; //const ll mod=1000000007; ll GCD(ll a,ll b) {return b ? GCD(b,a%b):a;} ll LCM(ll c,ll d){return c/GCD(c,d)*d;} struct __INIT{__INIT(){cin.tie(0);ios::sync_with_stdio(false);cout<<fixed<<setprecision(15);}} __init; template<class T> bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; } template<class T> bool chmin(T &a, const T &b) { if (a>b) { a=b; return 1; } return 0; } struct mint{ using ull=unsigned long long int; ull v; mint(ll vv=0){s(vv%mod+mod);} mint& s(ull vv){ v=vv<mod?vv:vv-mod; return *this; } //オーバーロード mint operator-()const{return mint()-*this;}//mint型にキャスト mint&operator+=(const mint&val){return s(v+val.v);} mint&operator-=(const mint&val){return s(v+mod-val.v);} mint&operator*=(const mint&val){ v=ull(v)*val.v%mod; return *this; } mint&operator/=(const mint&val){return *this*=val.inv();} mint operator+(const mint&val){return mint(*this)+=val;} mint operator-(const mint&val){return mint(*this)-=val;} mint operator*(const mint&val){return mint(*this)*=val;} mint operator/(const mint&val){return mint(*this)/=val;} mint pow(ll n)const{ mint res(1),x(*this); while(n){ if(n&1)res*=x; x*=x; n>>=1; } return res; } mint inv()const{return pow(mod-2);} //拡張ユークリッドの互除法 /* mint inv()const{ int x,y; int g=extgcd(v,mod,x,y); assert(g==1); if(x<0)x+=mod; return mint(x); }*/ friend ostream& operator<<(ostream&os,const mint&val){ return os<<val.v; }//出力 bool operator<(const mint&val)const{return v<val.v;} bool operator==(const mint&val)const{return v==val.v;} bool operator>(const mint&val)const{return v>val.v;} }; bool root[1000005]; int cnt[1000005]; mint val[1000005]; mint ans=0; V<int> g[1000005]; void dfs(int cur,int par){ if(cnt[cur]){ ans+=val[cur]; return; } for(int v:g[cur]){ if(v==par)continue; dfs(v,cur); cnt[cur]+=cnt[v]; val[cur]+=val[v]; } val[cur]+=mint(cnt[cur]); ans+=val[cur]; cnt[cur]++; return; } int main(){ int n; cin>>n; int a,b; for(int i=0;i<n-1;i++){ cin>>a>>b; g[a-1].push_back(b-1); root[b-1]=true; } for(int i=0;i<n;i++){ if(root[i])continue; dfs(i,-1); } cout<<ans<<"\n"; }