結果

問題 No.1483 Many Graph in Namori
コンテスト
ユーザー WTYAKIOI
提出日時 2024-04-30 15:10:25
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 109 ms / 2,000 ms
コード長 3,422 bytes
コンパイル時間 2,309 ms
コンパイル使用メモリ 188,652 KB
実行使用メモリ 27,008 KB
最終ジャッジ日時 2024-11-20 11:05:10
合計ジャッジ時間 8,488 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 56
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:80:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   80 | main()
      | ^~~~

ソースコード

diff # 

#pragma GCC optimize("Ofast","inline")
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=5e5+500,mod=998244353;

struct edge
{
    int u,v,nxt;
}e[N];

struct node
{
    int a,b;

    node operator+(node x)
    {
        return (node){(x.a+a)%mod,(x.b+b)%mod};
    }

    void mul(int x)
    {
        a=a*x%mod;b=b*x%mod;
    }
}f[N][3],s[N][3],las[N][2],w[N],lw[N];

node merge(node a,node b)
{
    return (node){(a.a*b.b%mod+a.b*b.a%mod)%mod,a.b*b.b%mod};
}

int n,k,head[N],cnt,in[N],inv1,inv2,ans,ro,vis[N],st[N],top,C;

int ksm(int a,int b)
{
    int ans=1;
    while(b)
    {
        if(b&1)ans=ans*a%mod;
        b>>=1;a=a*a%mod;
    }
    return ans;
}

void merge(node*f,node*h)
{
    node g[3];for(int i=0;i<3;i++)g[i]=f[i];
    for(int j=0;j<3;j++)
    for(int p=0;p<3;p++)
    {
        node re=merge(g[j],h[p]);
        if(j==1)re.mul(inv1);if(p==1)re.mul(inv1);
        int to=min(j+1,2ll);
        f[to]=f[to]+re;
    }
}

void dfs(int x,int fa)
{
    f[x][0]=(node){k*inv2%mod,k*inv2%mod};
    for(int i=head[x];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if(in[v]==2||v==fa)continue;
        dfs(v,x);merge(f[x],f[v]);
    }
    if(in[x]!=2)for(int i=0;i<3;i++)ans+=f[x][i].a;
}

void find(int x)
{
    vis[x]=1;st[++top]=x;
    for(int i=head[x];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if(in[v]==2&&!vis[v])return find(v);
    }
}

main()
{
    cin>>n>>k;
    inv1=ksm(k,mod-2),inv2=ksm(1-k,mod-2);C=ksm(1-k,n);
    for(int i=1;i<=n;i++)
    {
        int u,v;cin>>u>>v;++in[u],++in[v];
        e[++cnt]=(edge){u,v,head[u]};head[u]=cnt;
        e[++cnt]=(edge){v,u,head[v]};head[v]=cnt;
    }
    queue<int>q;
    for(int i=1;i<=n;i++)if(in[i]==1)q.push(i);
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for(int i=head[x];i;i=e[i].nxt)
        {
            int v=e[i].v;--in[v];
            if(in[v]==1)q.push(v);
        }
    }
    for(int i=1;i<=n;i++)if(in[i]==2)dfs(i,0),ro=i;
    find(ro);
    for(int i=1;i<=top;i++)for(int j=0;j<3;j++)s[i][j]=f[st[i]][j];
    for(int i=1;i<=top;i++)for(int j=0;j<3;j++)f[i][j]=s[i][j];
    for(int i=top;i>=1;i--)
    {
        if(i!=top)merge(s[i],s[i+1]);
        int val=0;
        for(int j=0;j<3;j++)val+=s[i][j].a;
        ans+=val;
    }
    node tmp;
    for(int i=1;i<=top;i++)
    {
        w[i]=f[i][0]+f[i][1];w[i].mul(inv1);
        w[i]=w[i]+f[i][2];
        if(i==1)tmp=w[i];else tmp=merge(tmp,w[i]);
    }
    ans=ans+tmp.a;
    for(int i=top;i>=1;i--)
    {
        las[i][0]=f[i][0];
        las[i][1]=f[i][1];las[i][1].mul(inv1);
        las[i][1]=las[i][1]+f[i][2];
        if(i==top)continue;
        node a=las[i][0],b=las[i][1];
        las[i][0]=merge(a,las[i+1][0]);las[i][0].mul(inv1);las[i][0]=las[i][0]+merge(a,las[i+1][1]);
        las[i][1]=merge(b,las[i+1][0]);las[i][1].mul(inv1);las[i][1]=las[i][1]+merge(b,las[i+1][1]);
    }
    for(int i=top;i>=1;i--)lw[i]=lw[i+1]+las[i][0]+las[i][1];
    for(int i=1;i<top;i++)
    {
        int val=lw[i+1].a;
        node now;
        now=f[i][1];now.mul(inv1);
        now=now+f[i][0]+f[i][2];
        if(i==1)
        {
            now=merge(now,lw[i+1]);
            ans+=now.a;
        }
        else
        {
            now=merge(now,tmp);
            now=merge(now,lw[i+1]);
            ans+=now.a;
        }
        if(i==1)tmp=w[i];
        else tmp=merge(tmp,w[i]);
    }
    cout<<(ans%mod*C%mod+mod)%mod;
}
0