結果

問題 No.1502 Many Simple Additions
コンテスト
ユーザー PCTprobability
提出日時 2021-05-07 22:45:21
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 4,487 bytes
コンパイル時間 26,588 ms
コンパイル使用メモリ 358,820 KB
最終ジャッジ日時 2025-01-21 08:33:41
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 2
other AC * 28 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
#include <unistd.h>
using namespace std;
#if __has_include(<atcoder/all>)
#include <atcoder/all>
using namespace atcoder;
#endif
using ll = long long;
#define REP3(i, m, n) for (int i = (m); (i) < int(n); ++ (i))
#define rep(i,a,b) for(int i=(int)(a);i<(int)(b);i++)
#define ALL(x) begin(x), end(x)
#define all(s) (s).begin(),(s).end()
//#define rep2(i, m, n) for (int i = (m); i < (n); ++i)
//#define rep(i, n) rep2(i, 0, n)
#define drep2(i, m, n) for (int i = (m)-1; i >= (n); --i)
#define drep(i, n) drep2(i, n, 0)
#define rever(vec) reverse(vec.begin(), vec.end())
#define sor(vec) sort(vec.begin(), vec.end())
#define fi first
#define se second
#define pb push_back
#define P pair<ll,ll>
#define REP(i, n) for (int i = 0; i < (n); ++i)
#define in scanner.read_int()
ll mod = 1000000007;
ll solve(ll XT,ll n,ll m,vector<ll> u,vector<ll> v,vector<ll> s){
    vector<vector<int>> E(n);
    vector<vector<long long>> S(n);
    for (int i=0; i<m; i++)
    {
        E[u[i]].push_back(v[i]);
        S[u[i]].push_back(s[i]);
        E[v[i]].push_back(u[i]);
        S[v[i]].push_back(s[i]);
    }
 
    vector<long long> V(n);
    vector<bool> X(n);
    vector<bool> F(n);
    int cv = -1;
    long long ca, cb;
 
    function<void(int,int,long long,bool)> BT = [&](int c, int p, long long v, bool x)
    {
        if (F[c] && X[c] != x)
        {
            cv = c;
            ca = V[c];
            cb = v;
        }
        if (F[c])
            return;
        F[c] = true;
        V[c] = v;
        X[c] = x;
        for (int i=0; i<(int)E[c].size(); i++)
        if (E[c][i] != p)
            BT(E[c][i], c, S[c][i]-v, !x);
    };
    BT(0, -1, 0LL, false);
    if (cv != -1)
    {
        if ((ca-cb)%2 != 0)
        {
          return 0;
        }
 
        F = vector<bool>(n);
        BT(cv, -1, (ca+cb)/2, false);
    }
 
    bool ok = true;
    bool bi = true;
 
    for (int i=0; i<n; i++)
    {
        for (int j=0; j<(int)E[i].size(); j++)
        {
            if (V[i]+V[E[i][j]] != S[i][j])
                ok = false;
            if (X[i] == X[E[i][j]])
                bi = false;
        }
    }
 
    if (!ok)
    {
      return 0;
    }
    if (!bi)
    {
        if (*min_element(V.begin(), V.end()) > 0&&*max_element(V.begin(), V.end()) <= XT)
        {
          return 1;
        }
        else{
          return 0;
        }
    }
 
    long long mn0 = XT;
    long long mn1 = 1;
 
    for (int i=0; i<n; i++)
    {
        if (X[i])
        {
          mn0 = min(mn0, V[i] + 1);
          mn1 = max(mn1,- XT + V[i]);
        }
        else
        {
            mn1 = max(mn1, V[i] + 1);
            mn0 = min(mn0, XT - V[i]);
        }
     // cout<<mn0<<" "<<mn1<<" "<<X[i]<<" "<<V[i]<<endl;
    }
 
    long long ans = max(mn0-mn1+1, 0LL);
 // cout<<ans<<endl;
    return ans;
}
class UnionFind{
public:
  vector<ll> par;
  vector<ll> siz;
  UnionFind(ll sz_):par(sz_),siz(sz_,1ll){
    for(int i=0;i<sz_;i++) par[i]=i;
  }
  void init(ll sz_){
    par.resize(sz_);
    siz.assign(sz_,1ll);
    for(int i=0;i<sz_;i++) par[i]=i;
  }
  ll root(ll x){
    while(par[x]!=x){
      x=par[x]=par[par[x]];
    }
    return x;
  }
  bool merge(ll x,ll y){
    x=root(x);
    y=root(y);
    if(x==y) return false;
    if(siz[x]<siz[y]) swap(x,y);
    siz[x]+=siz[y];
    par[y]=x;
    return true;
  }
  bool issame(ll x,ll y){
    return root(x)==root(y);
  }
  ll size(ll x){
    return siz[root(x)];
  }
};
int main(){
  ll n,m,K;
  cin>>n>>m>>K;
  vector<ll> a(m),b(m),c(m);
  UnionFind T(n);
  for(int i=0;i<m;i++){
    cin>>a[i]>>b[i]>>c[i];
    a[i]--;
    b[i]--;
    T.merge(a[i],b[i]);
  }
  vector<vector<ll>> p(n);
  for(int i=0;i<n;i++){
    p[T.root(i)].push_back(i);
  }
  vector<vector<ll>> h(n);
  for(int i=0;i<m;i++){
    h[T.root(a[i])].pb(i);
  }
  ll ans=1,ans2=1;
  for(int i=0;i<n;i++){
    vector<ll> u,v,s;
    map<ll,ll> M;
    for(int j=0;j<p[i].size();j++){
      M[p[i][j]]=j;
    }
    for(int j=0;j<h[i].size();j++){
      u.pb(M[a[h[i][j]]]);
      v.pb(M[b[h[i][j]]]);
      s.pb(c[h[i][j]]);
    }
    if(p[i].size()==0) continue;
    ans*=solve(K,p[i].size(),h[i].size(),u,v,s);
    ans%=mod;
    ans2*=solve(K-1,p[i].size(),h[i].size(),u,v,s);
    ans2%=mod;
   // cout<<solve(K,p[i].size(),h[i].size(),u,v,s)<<" "<<solve(K-1,p[i].size(),h[i].size(),u,v,s)<<endl;
  }
  cout<<((ans-ans2)%mod+mod)%mod<<endl;
}
0