結果

問題 No.1843 Tree ANDistance
コンテスト
ユーザー ___Johniel
提出日時 2022-02-19 01:56:50
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,010 ms / 2,000 ms
コード長 2,795 bytes
コンパイル時間 1,534 ms
コンパイル使用メモリ 177,824 KB
実行使用メモリ 10,136 KB
最終ジャッジ日時 2024-06-29 10:12:28
合計ジャッジ時間 21,205 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

// yukicoder/1843/main.cpp
// author: @___Johniel
// github: https://github.com/johniel/

#include <bits/stdc++.h>

#define each(i, c) for (auto& i : c)
#define unless(cond) if (!(cond))

using namespace std;

template<typename P, typename Q> ostream& operator << (ostream& os, pair<P, Q> p) { os << "(" << p.first << "," << p.second << ")"; return os; }
template<typename P, typename Q> istream& operator >> (istream& is, pair<P, Q>& p) { is >> p.first >> p.second; return is; }
template<typename T> ostream& operator << (ostream& os, vector<T> v) { os << "("; for (auto& i: v) os << i << ","; os << ")"; return os; }
template<typename T> istream& operator >> (istream& is, vector<T>& v) { for (auto& i: v) is >> i; return is; }
template<typename T> ostream& operator << (ostream& os, set<T> s) { os << "#{"; for (auto& i: s) os << i << ","; os << "}"; return os; }
template<typename K, typename V> ostream& operator << (ostream& os, map<K, V> m) { os << "{"; for (auto& i: m) os << i << ","; os << "}"; return os; }

template<typename T> inline T setmax(T& a, T b) { return a = std::max(a, b); }
template<typename T> inline T setmin(T& a, T b) { return a = std::min(a, b); }

using lli = long long int;
using ull = unsigned long long;
using point = complex<double>;
using str = string;
template<typename T> using vec = vector<T>;

constexpr array<int, 8> di({0, 1, -1, 0, 1, -1, 1, -1});
constexpr array<int, 8> dj({1, 0, 0, -1, 1, -1, -1, 1});
constexpr lli mod = 1e9 + 7;

class UnionFind {
public:
  UnionFind(int n) {
    r.resize(n, 0);
    p.resize(n, -1);
  }
  void unite(int a, int b) {
    a = find(a);
    b = find(b);
    if (a == b) return ;
    if (r[a] > r[b]) {
      p[a] += p[b];
      p[b] = a;
    } else {
      p[b] += p[a];
      p[a] = b;
      if(r[a] == r[b]) r[b]++;
    }
    return ;
  }
  int find(int a) {
    return (p[a] < 0) ? a : p[a] = find(p[a]);
  }
  bool same(int a, int b) {
    return find(a) == find(b);
  }
  size_t size(int n) {
    return -p.at(find(n));
  }
private:
  vector<int> r, p;
};

int main(int argc, char *argv[])
{
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  cout.setf(ios_base::fixed);
  cout.precision(15);

  int n;
  while (cin >> n) {
    int a[n], b[n], c[n];
    for (int i = 0; i < n - 1; ++i) {
      cin >> a[i] >> b[i] >> c[i];
      --a[i];
      --b[i];
    }

    lli z = 0;
    for (int i = 0; i < 50; ++i) {
      UnionFind uf(n);
      for (int j = 0; j < n - 1; ++j) {
        if (c[j] & (1LL << i)) uf.unite(a[j], b[j]);
      }
      lli w = (1LL << i) % mod;
      set<int> vis;
      for (int j = 0; j < n; ++j) {
        vis.insert(uf.find(j));
      }
      each (j, vis) {
        lli x = uf.size(j);
        (z += x * (x - 1) / 2 % mod * w) %= mod;
      }
    }
    cout << z << endl;
  }

  return 0;
}
0