結果

問題 No.1784 Not a star yet...
ユーザー vjudge1
提出日時 2025-06-08 17:20:59
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 67 ms / 2,000 ms
コード長 3,408 bytes
コンパイル時間 2,134 ms
コンパイル使用メモリ 174,040 KB
実行使用メモリ 21,248 KB
最終ジャッジ日時 2025-06-08 17:21:05
合計ジャッジ時間 6,205 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 61
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 260,mod = 998244353;
typedef array<int,N> vec;
int addv(int x,int y){return (x+y>=mod)?(x+y-mod):(x+y);}
int subv(int x,int y){return (x>=y)?(x-y):(x-y+mod);}
void add(int& x,int y){x = addv(x,y);}
void sub(int& x,int y){x = subv(x,y);}
int qpow(int a,int n){
    int res = 1; a%=mod;
    while(n){
        if(n&1)res = 1ll*res*a%mod;
        n>>=1; a = 1ll*a*a%mod;
    }
    return res;
}
int qinv(int a){return qpow(a,mod-2);}
//
int n,m,U,Cv,p[N],q[N],cp,cq,ans;

int A[N][N],B[N][N],C[N][N],D[N][N],E[N][N],v[N][N];
vec g[N][N];

int F(int x,int y){return 1ll*x*qinv(y)%mod;}

vec operator+(vec x,vec y){
    vec z = {};
    for(int i=0;i<=cq+1;i++)z[i] = addv(x[i],y[i]);
    return z;
}
vec operator*(int x,vec y){
    vec z = {};
    for(int i=0;i<=cq+1;i++)z[i] = 1ll*x*y[i]%mod;
    return z;
}

int a[N][N],x[N],w[N][N];
void solve(int r){
    if(r > m)return;
    int x = -1; for(int i=r;i<=m;i++)if(a[i][r]){x = i; break;}
    for(int i=r;i<=m+1;i++)swap(a[r][i],a[x][i]);
    for(int i=r+1;i<=m;i++)if(a[i][r]){
        int w = 1ll*a[i][r]*qinv(a[r][r])%mod;
        for(int j=r;j<=m+1;j++)sub(a[i][j],1ll*a[r][j]*w%mod);
    }
    solve(r+1);
}

int main(){ 
    cin>>n; 
    for(int i=1,u,v,l;i<n;i++){
        cin>>u>>v>>l;
        if(l==1)p[u]++,p[v]++,cp++; else q[u]++,q[v]++,cq++;
    }
    Cv = cp+2*cq; U = n*(n-1)/2 - (n-2);
    for(int i=0;i<=cp;i++)for(int j=0;j<=cq;j++){
        v[i][j] = F(i+2*j,2*Cv);
        {   //move G1
            add(B[i][j],1ll*F(cp-i,Cv)*F(U-(n-1-(i+j)),U)%mod);
            add(A[i][j],1ll*F(i,Cv)*F(U-(n-1-(i+j-1)),U)%mod);
            add(B[i][j],1ll*F(i,Cv)*F(n-1-(i+j-1),U)%mod);
            add(C[i][j],1ll*F(cp-i,Cv)*F(n-1-(i+j),U)%mod);
        }
        {   //move G2
            add(B[i][j],1ll*F(2*(cq-j),Cv)*F(U-(n-1-(i+j)),U)%mod);
            add(D[i][j],1ll*F(2*j,Cv)*F(U-(n-1-(i+j-1)),U)%mod);
            add(B[i][j],1ll*F(2*j,Cv)*F(n-1-(i+j-1),U)%mod);
            add(E[i][j],1ll*F(2*(cq-j),Cv)*F(n-1-(i+j),U)%mod);
        }
        sub(B[i][j],1);
    }
    //??
    for(int j=0;j<=cq;j++)g[0][j][j] = 1;
    for(int i=0;i<cp;i++)for(int j=0;j<=cq;j++){
        vec tmp = {};
        if(i)tmp = tmp + A[i][j] * g[i-1][j];
        tmp = tmp + B[i][j] * g[i][j];
        if(j)tmp = tmp + D[i][j] * g[i][j-1];
        if(j<cq)tmp = tmp + E[i][j] * g[i][j+1];
        add(tmp[cq+1],v[i][j]);
        g[i+1][j] = qinv(mod-C[i][j]) * tmp;
    }
    for(int j=0;j<=cq;j++){
        vec tmp = {};
        if(cp)tmp = tmp + A[cp][j] * g[cp-1][j];
        tmp = tmp + B[cp][j] * g[cp][j];
        if(j)tmp = tmp + D[cp][j] * g[cp][j-1];
        if(j<cq)tmp = tmp + E[cp][j] * g[cp][j+1];
        add(tmp[cq+1],v[cp][j]);

        for(int k=0;k<=cq;k++)a[j+1][k+1] = tmp[k];
        sub(a[j+1][cq+2],tmp[cq+1]);
    }
    m = cq+1;
    solve(1);
    for(int i=m;i>=1;i--){
        x[i] = 1ll*a[i][m+1]*qinv(a[i][i])%mod;
        for(int j=i-1;j>=1;j--)sub(a[j][m+1],1ll*x[i]*a[j][i]%mod);
    }
    //
    for(int i=0;i<=cp;i++)for(int j=0;j<=cq;j++){
        for(int k=0;k<=cq;k++)add(w[i][j],1ll*g[i][j][k]*x[k+1]%mod);
        add(w[i][j],g[i][j][cq+1]);
    }
    for(int i=1;i<=n;i++)add(ans,w[p[i]][q[i]]);
    sub(ans,w[cp][cq]);
    for(int i=1;i<=cp;i++)sub(ans,w[1][0]);
    for(int i=1;i<=cq;i++)sub(ans,w[0][1]);

    cout<<ans<<endl;
    return 0;
}
0