結果

問題 No.1507 Road Blocked
ユーザー hiikunZhiikunZ
提出日時 2021-05-14 23:23:47
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 105 ms / 2,000 ms
コード長 1,327 bytes
コンパイル時間 1,889 ms
コンパイル使用メモリ 198,664 KB
最終ジャッジ日時 2025-01-21 12:17:47
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll = long long int;
using ld = long double;
#define pow(n,m) powl(n,m)
#define sqrt(n) sqrtl(n)
const ll MAX = 1000000000000000000;
const ll MOD = 998244353;
void randinit(){srand((unsigned)time(NULL));}
ll modinv(ll A,ll M){
    //A*r+B*n=gcd(A,B)
    A %= M;
    if(A == 0 || __gcd(A,M) != 1){
        //cout << "Error modinv(" << A << "," << M << ")" << endl;
        return -1;
    }
    ll B = M,U = 1,V = 0;
    while(B){
        ll T = A / B;
        A -= T * B;
        swap(A,B);
        U -= T * V;
        swap(U,V);
    }
    U %= M;
    if(U < 0) U += M;
    return U;
}
ll N;
vector<vector<ll>> G(100000);
vector<ll> S(100000,0);
ll s = 0;
ll dfs(ll n){
    if(S[n] == 1) return 0;
    S[n] = 1;
    ll ans = 1;
    for(ll i = 0;i < G[n].size();i++) ans += dfs(G[n][i]);
    if(n != 0) s += (ans * (ans - 1));
    if(n != 0)s += ((N - ans) * (N - ans - 1));
    s %= MOD;
    return ans;
}
int main(){
    cin >> N;
    for(ll i = 0;i < N - 1;i++){
        ll a,b;
        cin >> a >> b;
        G[a - 1].emplace_back(b - 1);
        G[b - 1].emplace_back(a - 1);
    }
    dfs(0);
    ll t = (N * (N - 1)) % MOD;
    t = (t * (N - 1)) % MOD;
    s = (s * modinv(2,MOD)) % MOD;
    t = (t * modinv(2,MOD)) % MOD;
    cout << (s * modinv(t,MOD)) % MOD << endl;
}
0