Submission #1049077

#	Time	Username	Problem	Language	Result	Execution time	Memory
1049077		NeroZein	Transport (COCI19_transport)	C++17	26 / 130	1072 ms	12628 KiB

Main

#include "bits/stdc++.h"
using namespace std;

#ifdef Nero
#include "Deb.h"
#else
#define debug(...)
#endif

const int N = 1e5 + 5;

int a[N];
vector<pair<int, int>> g[N];

int dfs(int v, int p, long long fuel) {
  if (fuel < 0) {
    return -1; 
  }
  int ret = 0;
  fuel += a[v];
  for (auto [u, w] : g[v]) {
    if (u == p) {
      continue; 
    }
    ret += 1 + dfs(u, v, fuel - w); 
  }
  return ret; 
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int n;
  cin >> n;
  for (int i = 1; i <= n; ++i) {
    cin >> a[i];
  }
  for(int i = 0; i < n - 1; ++i) {
    int u, v, w;
    cin >> u >> v >> w;
    g[u].push_back({v, w});
    g[v].push_back({u, w});
  }
  long long ans = 0; 
  for (int i = 1; i <= n; ++i) {
    ans += dfs(i, i, 0);    
  }
  cout << ans << '\n';
  return 0;
}

• Subtask 1: 13 / 13
• Subtask 2: 13 / 13
• Subtask 3: 0 / 13
• Subtask 4: 0 / 13
• Subtask 5: 0 / 13
• Subtask 6: 0 / 13
• Subtask 7: 0 / 13
• Subtask 8: 0 / 13
• Subtask 9: 0 / 13
• Subtask 10: 0 / 13