Submission #44819127

---

Source Code Expand

#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
#define Ed for(int i=h[x];~i;i=ne[i])
#define Ls(i,l,r) for(int i=l;i<r;++i)
#define Rs(i,l,r) for(int i=l;i>r;--i)
#define Le(i,l,r) for(int i=l;i<=r;++i)
#define Re(i,l,r) for(int i=l;i>=r;--i)
#define L(i,l) for(int i=0;i<l;++i)
#define E(i,l) for(int i=1;i<=l;++i)
#define W(t) while(t--)
#define Wh while
typedef long long ll;
#define add(a,b) (a+=b)>=mod&&(a-=mod)
#define sub(a,b) (a-=b)<0&&(a+=mod)

const int N=200010,mod=998244353;
int n,f[N],gans[N],ans[N],sz[N];
vector<int>g[N];
int find(int x){
    if(f[x]==x)return x;
    return f[x]=find(f[x]);
}
ll qpow(ll a,int b=mod-2){
    ll res=1;
    while(b){
        if(b&1)res=res*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return res;
}
int main(){
    #ifndef ONLINE_JUDGE
    freopen("1.in","r",stdin);
    #endif
    scanf("%d",&n);
    E(i, n)f[i]=i,sz[i]=1,g[i].push_back(i);
    E(i, n-1){
        int a,b;
        scanf("%d%d",&a,&b);
        a=find(a),b=find(b);
        if(sz[a]>sz[b])swap(a,b);
        //merge a to b
        ll inv=qpow(sz[a]+sz[b]);
        int newv=sz[a]*inv%mod,newv2=sz[b]*inv%mod,old=newv2;
        // printf("newv=%d/%d,newv2=%d/%d\n",sz[a],(sz[a]+sz[b]),sz[b],(sz[a]+sz[b]));
        add(old,gans[b]);
        for(int v:g[a]){
            add(ans[v],gans[a]),add(ans[v],newv),sub(ans[v],old);
            g[b].push_back(v);
        }
        gans[a]=0;
        gans[b]=old;
        newv2=old;
        f[a]=b;
        sz[b]+=sz[a];
        sz[a]=0;
        vector<int>().swap(g[a]);
    }
    E(j, n)
        if(sz[j]){
            E(i, n)add(ans[i],gans[j]);
            break;
        }
    E(i, n)printf("%d ",ans[i]);
    return 0;
}

Submission Info

Compile Error

Main.cpp: In function ‘int main()’:
Main.cpp:38:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   38 |     scanf("%d",&n);
      |     ~~~~~^~~~~~~~~
Main.cpp:42:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   42 |         scanf("%d%d",&a,&b);
      |         ~~~~~^~~~~~~~~~~~~~

Judge Result