提出 #3610561

---

ソースコード 拡げる

#include <algorithm>
#include <iostream>
#include <map>
#include <vector>
using namespace std;

const int MAX = 2540;
const int N = 1e5;
vector<pair<int, int>> path[N];
map<int, int> cnt[N];

int main() {
  ios::sync_with_stdio(false);
  int n;
  cin >> n;
  int m;
  cin >> m;
  for (int i = 0; i < m; ++i) {
    int a, b, l;
    cin >> a >> b >> l;
    --a;
    --b;
    path[a].emplace_back(l, b);
    path[b].emplace_back(l, a);
    ++cnt[a][l];
    ++cnt[b][l];
  }
  long long res = 0;
  for (int mid = 0; mid < n; ++mid) {
    for (const auto &a_path : path[mid]) {
      int len = a_path.first;
      int u = a_path.second;
      res += cnt[mid][MAX - len];
      if (len + len == MAX) --res;
    }
  }
  cout << res / 2 << '\n';
  return 0;
}

提出情報

ジャッジ結果