結果
問題 | No.1483 Many Graph in Namori |
ユーザー | beet |
提出日時 | 2021-03-15 17:44:59 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 241 ms / 2,000 ms |
コード長 | 4,829 bytes |
コンパイル時間 | 3,148 ms |
コンパイル使用メモリ | 235,584 KB |
実行使用メモリ | 24,448 KB |
最終ジャッジ日時 | 2024-11-08 01:41:26 |
合計ジャッジ時間 | 11,682 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 2 ms
5,248 KB |
testcase_03 | AC | 2 ms
5,248 KB |
testcase_04 | AC | 2 ms
5,248 KB |
testcase_05 | AC | 2 ms
5,248 KB |
testcase_06 | AC | 2 ms
5,248 KB |
testcase_07 | AC | 2 ms
5,248 KB |
testcase_08 | AC | 2 ms
5,248 KB |
testcase_09 | AC | 2 ms
5,248 KB |
testcase_10 | AC | 141 ms
15,616 KB |
testcase_11 | AC | 66 ms
10,496 KB |
testcase_12 | AC | 94 ms
12,928 KB |
testcase_13 | AC | 84 ms
11,776 KB |
testcase_14 | AC | 104 ms
13,184 KB |
testcase_15 | AC | 206 ms
18,688 KB |
testcase_16 | AC | 65 ms
10,368 KB |
testcase_17 | AC | 24 ms
6,400 KB |
testcase_18 | AC | 114 ms
14,080 KB |
testcase_19 | AC | 111 ms
13,696 KB |
testcase_20 | AC | 68 ms
10,624 KB |
testcase_21 | AC | 28 ms
6,784 KB |
testcase_22 | AC | 114 ms
14,336 KB |
testcase_23 | AC | 128 ms
15,104 KB |
testcase_24 | AC | 67 ms
10,368 KB |
testcase_25 | AC | 69 ms
10,880 KB |
testcase_26 | AC | 194 ms
19,328 KB |
testcase_27 | AC | 221 ms
20,224 KB |
testcase_28 | AC | 32 ms
7,168 KB |
testcase_29 | AC | 76 ms
11,392 KB |
testcase_30 | AC | 219 ms
20,608 KB |
testcase_31 | AC | 241 ms
20,736 KB |
testcase_32 | AC | 229 ms
20,608 KB |
testcase_33 | AC | 232 ms
20,608 KB |
testcase_34 | AC | 220 ms
20,608 KB |
testcase_35 | AC | 70 ms
10,112 KB |
testcase_36 | AC | 80 ms
11,136 KB |
testcase_37 | AC | 157 ms
16,384 KB |
testcase_38 | AC | 29 ms
6,400 KB |
testcase_39 | AC | 216 ms
20,224 KB |
testcase_40 | AC | 70 ms
10,880 KB |
testcase_41 | AC | 174 ms
19,712 KB |
testcase_42 | AC | 128 ms
15,872 KB |
testcase_43 | AC | 164 ms
18,816 KB |
testcase_44 | AC | 191 ms
20,736 KB |
testcase_45 | AC | 9 ms
5,248 KB |
testcase_46 | AC | 56 ms
9,856 KB |
testcase_47 | AC | 195 ms
19,456 KB |
testcase_48 | AC | 211 ms
20,480 KB |
testcase_49 | AC | 229 ms
20,992 KB |
testcase_50 | AC | 175 ms
20,992 KB |
testcase_51 | AC | 175 ms
21,248 KB |
testcase_52 | AC | 169 ms
20,096 KB |
testcase_53 | AC | 169 ms
20,992 KB |
testcase_54 | AC | 171 ms
20,992 KB |
testcase_55 | AC | 172 ms
21,120 KB |
testcase_56 | AC | 136 ms
19,712 KB |
testcase_57 | AC | 106 ms
19,840 KB |
testcase_58 | AC | 140 ms
24,448 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; using Int = long long; const char newl = '\n'; template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;} template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;} template<typename T> void drop(const T &x){cout<<x<<endl;exit(0);} template<typename T=int> vector<T> read(size_t n){ vector<T> ts(n); for(size_t i=0;i<n;i++) cin>>ts[i]; return ts; } template<typename T, T MOD = 1000000007> struct Mint{ static constexpr T mod = MOD; T v; Mint():v(0){} Mint(signed v):v(v){} Mint(long long t){v=t%MOD;if(v<0) v+=MOD;} Mint pow(long long k){ Mint res(1),tmp(v); while(k){ if(k&1) res*=tmp; tmp*=tmp; k>>=1; } return res; } static Mint add_identity(){return Mint(0);} static Mint mul_identity(){return Mint(1);} Mint inv(){return pow(MOD-2);} Mint& operator+=(Mint a){v+=a.v;if(v>=MOD)v-=MOD;return *this;} Mint& operator-=(Mint a){v+=MOD-a.v;if(v>=MOD)v-=MOD;return *this;} Mint& operator*=(Mint a){v=1LL*v*a.v%MOD;return *this;} Mint& operator/=(Mint a){return (*this)*=a.inv();} Mint operator+(Mint a) const{return Mint(v)+=a;} Mint operator-(Mint a) const{return Mint(v)-=a;} Mint operator*(Mint a) const{return Mint(v)*=a;} Mint operator/(Mint a) const{return Mint(v)/=a;} Mint operator-() const{return v?Mint(MOD-v):Mint(v);} bool operator==(const Mint a)const{return v==a.v;} bool operator!=(const Mint a)const{return v!=a.v;} bool operator <(const Mint a)const{return v <a.v;} static Mint comb(long long n,int k){ Mint num(1),dom(1); for(int i=0;i<k;i++){ num*=Mint(n-i); dom*=Mint(i+1); } return num/dom; } }; template<typename T, T MOD> constexpr T Mint<T, MOD>::mod; template<typename T, T MOD> ostream& operator<<(ostream &os,Mint<T, MOD> m){os<<m.v;return os;} template<typename F> struct FixPoint : F{ FixPoint(F&& f):F(forward<F>(f)){} template<typename... Args> decltype(auto) operator()(Args&&... args) const{ return F::operator()(*this,forward<Args>(args)...); } }; template<typename F> inline decltype(auto) MFP(F&& f){ return FixPoint<F>{forward<F>(f)}; } //INSERT ABOVE HERE signed main(){ cin.tie(0); ios::sync_with_stdio(0); int n,k; cin>>n>>k; vector<set<int>> S(n); for(int i=0;i<n;i++){ int a,b; cin>>a>>b; a--;b--; S[a].emplace(b); S[b].emplace(a); } vector<vector<int>> T(n); queue<int> que; for(int i=0;i<n;i++) if(S[i].size()==1) que.emplace(i); while(!que.empty()){ int v=que.front();que.pop(); int p=*S[v].begin(); T[p].emplace_back(v); S[p].erase(v); S[v].erase(p); if(S[p].size()==1) que.emplace(p); } vector<int> in_loop=[&](){ int pos=-1; for(int i=0;i<n;i++) if(S[i].size()>0) pos=i; assert(~pos); vector<int> res; while(not S[pos].empty()){ res.emplace_back(pos); int nxt=*S[pos].begin(); S[pos].erase(nxt); S[nxt].erase(pos); pos=nxt; } return res; }(); using M = Mint<int, 998244353>; M ans{0}; vector<M> dp(n); vector<int> sz(n); M inv=(M(1)-M(k)).inv(); // not use loop for(int r:in_loop){ MFP([&](auto dfs,int v)->void{ sz[v]=1; for(int u:T[v]){ dfs(u); sz[v]+=sz[u]; } dp[v]=inv.pow(sz[v])-M(1); for(int u:T[v]) dp[v]+=dp[u]*inv.pow(sz[v]-sz[u]); // see from descents for(int u:T[v]) ans+=(inv.pow(sz[v]-sz[u])-M(1))*dp[u]; // see from v M zero=1; M one=0; M all=inv.pow(sz[v]-1); for(int u:T[v]) one+=inv.pow(sz[u])-M(1); ans+=M(k)*inv*zero; ans+=M(k)*inv*one; ans+=inv*(all-(zero+one)); })(r); } auto calc=MFP([&](auto dfs,int r,int v)->M{ M res=inv.pow(sz[v])-M(1); if(v==r) res+=M(1); for(int u:T[v]) res+=dfs(r,u)*inv.pow(sz[v]-sz[u]); return res; }); // use all edges in loop for(int r:in_loop) ans+=calc(r,r)*inv.pow(n-sz[r]); // use partial edges in loop // prepare int num=0; M sum{0}; M uku{0}; auto proceed=[&](int r){ sum*=inv.pow(sz[r]); uku*=inv.pow(sz[r]); sum+=calc(r,r)*inv.pow(num+n); uku-=M(1); num+=sz[r]; }; for(int r:in_loop) proceed(r); assert(num==n); // add for(int r:in_loop){ // see from r's subtree ans+=MFP([&](auto dfs,int v)->M{ // OK M res=inv.pow(sz[v])-M(1); for(int u:T[v]) res+=dfs(u)*inv.pow(sz[v]-sz[u]); return res; })(r)*(inv.pow(n-sz[r])-M(1)); // see from left sum-=calc(r,r)*inv.pow(num+n-n)*inv.pow(n-sz[r]); uku+=M(1)*inv.pow(n-sz[r]); ans+=(sum/inv.pow(num+sz[r])+uku)*(inv.pow(sz[r])-M(1)); proceed(r); } ans/=inv.pow(n); cout<<ans<<newl; return 0; }