Submission #5565976

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);
  
  map<int, map<int, int>> m;
  int N, M;
  cin >> N >> M;
  vector<map<int, int>> v(N);
  for (int i = 0; i < M; i++) {
    int A, B, L;
    cin >> A >> B >> L;
    m[A][L]++;
    m[B][L]++;
    v[A - 1][L]++;
    v[B - 1][L]++;
  }
  int ans = 0;
  for (int i = 0; i < N; i++) {
    for (auto j = v[i].begin(); j != v[i].end(); j++) {
      int l = j -> first;
      int t1 = j -> second;
      int t2 = v[i][2540 - l];
      if (l == 1270) t1--;
      ans += t1 * t2;
    }
  }
  ans /= 2;
  cout << ans << "\n";
  return 0;
}

Submission Info

Judge Result