結果

問題 No.1483 Many Graph in Namori
ユーザー WTYAKIOIWTYAKIOI
提出日時 2024-04-30 15:10:25
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 108 ms / 2,000 ms
コード長 3,422 bytes
コンパイル時間 1,980 ms
コンパイル使用メモリ 189,232 KB
実行使用メモリ 40,052 KB
最終ジャッジ日時 2024-04-30 15:10:34
合計ジャッジ時間 7,599 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
19,916 KB
testcase_01 AC 5 ms
19,920 KB
testcase_02 AC 5 ms
20,048 KB
testcase_03 AC 5 ms
19,924 KB
testcase_04 AC 5 ms
19,796 KB
testcase_05 AC 4 ms
19,920 KB
testcase_06 AC 5 ms
19,924 KB
testcase_07 AC 5 ms
19,924 KB
testcase_08 AC 5 ms
19,916 KB
testcase_09 AC 4 ms
19,920 KB
testcase_10 AC 57 ms
26,872 KB
testcase_11 AC 33 ms
24,484 KB
testcase_12 AC 44 ms
24,636 KB
testcase_13 AC 40 ms
24,564 KB
testcase_14 AC 47 ms
26,712 KB
testcase_15 AC 71 ms
29,120 KB
testcase_16 AC 33 ms
22,560 KB
testcase_17 AC 16 ms
19,996 KB
testcase_18 AC 51 ms
26,640 KB
testcase_19 AC 47 ms
26,744 KB
testcase_20 AC 33 ms
24,604 KB
testcase_21 AC 18 ms
22,248 KB
testcase_22 AC 49 ms
26,664 KB
testcase_23 AC 68 ms
26,844 KB
testcase_24 AC 33 ms
24,664 KB
testcase_25 AC 37 ms
24,508 KB
testcase_26 AC 79 ms
31,416 KB
testcase_27 AC 77 ms
31,264 KB
testcase_28 AC 20 ms
22,224 KB
testcase_29 AC 39 ms
24,556 KB
testcase_30 AC 82 ms
29,244 KB
testcase_31 AC 81 ms
29,376 KB
testcase_32 AC 86 ms
31,296 KB
testcase_33 AC 81 ms
29,496 KB
testcase_34 AC 80 ms
29,240 KB
testcase_35 AC 43 ms
28,736 KB
testcase_36 AC 42 ms
26,656 KB
testcase_37 AC 69 ms
37,504 KB
testcase_38 AC 18 ms
20,200 KB
testcase_39 AC 90 ms
39,948 KB
testcase_40 AC 33 ms
24,588 KB
testcase_41 AC 70 ms
31,560 KB
testcase_42 AC 50 ms
27,092 KB
testcase_43 AC 66 ms
29,292 KB
testcase_44 AC 74 ms
29,456 KB
testcase_45 AC 9 ms
22,060 KB
testcase_46 AC 32 ms
26,772 KB
testcase_47 AC 79 ms
36,040 KB
testcase_48 AC 108 ms
36,112 KB
testcase_49 AC 87 ms
35,584 KB
testcase_50 AC 80 ms
30,240 KB
testcase_51 AC 74 ms
29,472 KB
testcase_52 AC 76 ms
31,512 KB
testcase_53 AC 79 ms
33,692 KB
testcase_54 AC 75 ms
31,660 KB
testcase_55 AC 79 ms
30,116 KB
testcase_56 AC 74 ms
40,052 KB
testcase_57 AC 84 ms
37,364 KB
testcase_58 AC 68 ms
31,496 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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