結果

問題 No.1483 Many Graph in Namori
ユーザー WTYAKIOIWTYAKIOI
提出日時 2024-04-30 15:10:25
言語 C++14
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
19,924 KB
testcase_01 AC 4 ms
19,920 KB
testcase_02 AC 4 ms
19,920 KB
testcase_03 AC 4 ms
19,916 KB
testcase_04 AC 5 ms
19,916 KB
testcase_05 AC 5 ms
19,916 KB
testcase_06 AC 5 ms
19,792 KB
testcase_07 AC 5 ms
19,916 KB
testcase_08 AC 5 ms
19,916 KB
testcase_09 AC 5 ms
19,792 KB
testcase_10 AC 58 ms
23,008 KB
testcase_11 AC 35 ms
22,436 KB
testcase_12 AC 44 ms
24,512 KB
testcase_13 AC 40 ms
24,568 KB
testcase_14 AC 48 ms
9,984 KB
testcase_15 AC 75 ms
13,824 KB
testcase_16 AC 34 ms
8,320 KB
testcase_17 AC 14 ms
5,760 KB
testcase_18 AC 50 ms
10,752 KB
testcase_19 AC 49 ms
10,624 KB
testcase_20 AC 36 ms
8,448 KB
testcase_21 AC 18 ms
6,016 KB
testcase_22 AC 52 ms
10,880 KB
testcase_23 AC 59 ms
11,648 KB
testcase_24 AC 35 ms
8,320 KB
testcase_25 AC 37 ms
8,704 KB
testcase_26 AC 78 ms
14,464 KB
testcase_27 AC 85 ms
14,720 KB
testcase_28 AC 20 ms
6,272 KB
testcase_29 AC 40 ms
9,216 KB
testcase_30 AC 83 ms
15,232 KB
testcase_31 AC 90 ms
15,104 KB
testcase_32 AC 86 ms
15,360 KB
testcase_33 AC 90 ms
14,976 KB
testcase_34 AC 89 ms
15,232 KB
testcase_35 AC 38 ms
12,800 KB
testcase_36 AC 46 ms
14,208 KB
testcase_37 AC 88 ms
21,632 KB
testcase_38 AC 19 ms
7,552 KB
testcase_39 AC 109 ms
27,008 KB
testcase_40 AC 33 ms
8,576 KB
testcase_41 AC 75 ms
14,592 KB
testcase_42 AC 53 ms
11,776 KB
testcase_43 AC 69 ms
13,824 KB
testcase_44 AC 80 ms
15,104 KB
testcase_45 AC 6 ms
5,248 KB
testcase_46 AC 33 ms
10,240 KB
testcase_47 AC 89 ms
19,692 KB
testcase_48 AC 94 ms
20,928 KB
testcase_49 AC 95 ms
21,376 KB
testcase_50 AC 87 ms
16,000 KB
testcase_51 AC 87 ms
15,872 KB
testcase_52 AC 86 ms
15,744 KB
testcase_53 AC 84 ms
16,000 KB
testcase_54 AC 84 ms
15,872 KB
testcase_55 AC 87 ms
15,872 KB
testcase_56 AC 85 ms
24,960 KB
testcase_57 AC 81 ms
20,992 KB
testcase_58 AC 71 ms
17,024 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