結果

問題 No.1207 グラフX
ユーザー hs0319boy
提出日時 2020-08-30 15:42:02
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 655 ms / 2,000 ms
コード長 2,484 bytes
コンパイル時間 2,108 ms
コンパイル使用メモリ 181,256 KB
実行使用メモリ 42,368 KB
最終ジャッジ日時 2024-11-15 09:59:35
合計ジャッジ時間 25,648 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using ll=long long;
using vin=vector<int>;
using vll=vector<long long>;
using vvin=vector<vector<int>>;
using vvll=vector<vector<long long>>;
using vstr=vector<string>;
using vvstr=vector<vector<string>>;
using vch=vector<char>;
using vvch=vector<vector<char>>;
using vbo=vector<bool>;
using vvbo=vector<vector<bool>>;
using vpii=vector<pair<int,int>>;
using pqsin=priority_queue<int,vector<int>,greater<int>>;
#define mp make_pair
#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define rep2(i,s,n) for(int i=(s);i<(int)(n);i++)
#define all(v) v.begin(),v.end()
#define decp(n) cout<<fixed<<setprecision((int)n)
const ll inf=1e9+7;
const ll INF=1e18;
const ll mod=998244353;

int MAX_N=2e5+20;

vin par(MAX_N);
vin deep(MAX_N);
vin sizee(MAX_N);
void init(int n);
int root(int x);
void unite(int x,int y);
bool same(int x,int y);
int group_size(int x);

vvin baby(MAX_N);
vin babynum(MAX_N,1);

void babycount(int v,int p){//自身を含む子孫の数
    if(baby[v].size()==0)return;
    for(auto b:baby[v]){
        if(b==p)continue;
        babycount(b,v);
        babynum[v]+=babynum[b];
    }
}

ll modpow(ll a,ll n,ll m=inf){
    ll res=1;
    while(n){
        if(n&1)res=res*a%m;
        a=a*a%m;
        n>>=1;
    }
    return res;
}

ll fermat(ll a,ll b,ll m){
    return (a%m)*modpow(b,m-2,m)%m;
}

int main(){
    ll n,m,X;cin>>n>>m>>X;
    init(n);
    map<pair<int,int>,ll> road;
    int x,y;ll z;
    rep(i,m){
        cin>>x>>y>>z;
        if(same(x,y))continue;
        unite(x,y);
        road[mp(min(x,y),max(x,y))]=z;
        baby[x].push_back(y);
        baby[y].push_back(x);
    }
    babycount(1,-1);
    ll ans=(ll)0,tmp;
    rep2(i,1,n+1){
        for(auto b:baby[i]){
            tmp=min(babynum[i],babynum[b]);
            ans+=modpow(X,road[mp(min(i,b),max(i,b))])*((tmp*(n-tmp))%inf);
            ans%=inf;
        }
    }
    cout<<fermat(ans,(ll)2,inf)<<endl;
}

void init(int n){
    rep(i,n+1){
        par[i]=i;
        deep[i]=0;
        sizee[i]=1;
    }
}
int root(int x){
    if(par[x]==x)return x;
    else return par[x]=root(par[x]);
}
void unite(int x,int y){
    x=root(x);
    y=root(y);
    if(x==y)return;
    if(deep[x]<deep[y]){
        par[x]=y;
        sizee[y]+=sizee[x];
    }
    else{
        par[y]=x;
        sizee[x]+=sizee[y];
        if(deep[x]==deep[y])deep[x]++;
    }
}
bool same(int x,int y){
    return root(x)==root(y);
}
int group_size(int x){
    return sizee[root(x)];
}
0