oj.uz

Submission #883819

# Submission time Handle Problem Language Result Execution time Memory
883819 2023-12-06T06:47:55 Z vjudge1 Putovanje (COCI20_putovanje) C++14
0 / 110
 1000 ms 38068 KB 
#include <bits/stdc++.h>
using namespace std;
#define sp << " " << 
#define int long long
#define vi vector<int>
#define F(xxx,yyy) for (int xxx=1;xxx<=yyy;xxx++)
#define pii pair<int,int>
const int N = 2e5+1;
vi edges[N],v(N),w(N),pass(N,0);
set<int> sts[N];
map<pii,int> edg;
int n;
void dfs(int node,int p) {
  for (auto it : edges[node]) if (it != p) dfs(it,node);
  for (auto it : edges[node]) {
    if (it == p) continue;
    for (auto itt : sts[it]) {
      int c = 2;
      c-=(itt==1 || sts[it].count(itt-1));
      c-=(itt==n || sts[it].count(itt+1));
      pass[edg[{node,it}]]+=c;
    }
  }
  sts[node].insert(node);
  for (auto it : edges[node]) {
    if (it == p) continue;
    if (sts[it].size() > sts[node].size()) sts[node].swap(sts[it]);
    for (auto itt : sts[it]) sts[node].insert(it);
  }
}


void solve() {
  cin >> n;
  F(i,n-1) {
    int a,b;
    cin >> a >> b >> v[i] >> w[i];
    edges[a].push_back(b);
    edges[b].push_back(a);
    edg[{a,b}] = edg[{b,a}] = i;
  }
  dfs(1,1);
  int ans = 0;
  F(i,n-1) {
    if (pass[i]*v[i] <= w[i]) ans+=pass[i]*v[i];
    else ans+=w[i];
  }
  cout << ans << endl;
}
    
                  
                             
signed main() { 
  ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
  #ifdef Local
  freopen("in","r",stdin);
  freopen("out","w",stdout); 
  #endif
  int t = 1;
  //cin >> t;
	F(i,t) solve();
}

Compilation message

putovanje.cpp: In function 'void dfs(long long int, long long int)':
putovanje.cpp:28:15: warning: unused variable 'itt' [-Wunused-variable]
   28 |     for (auto itt : sts[it]) sts[node].insert(it);
      |               ^~~

Subtask #1
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 19036 KB Output is correct
2 Incorrect 8 ms 19548 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 25.0

# Verdict Execution time Memory Grader output
1 Execution timed out 1033 ms 38068 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #3
0 / 65.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 19036 KB Output is correct
2 Incorrect 8 ms 19548 KB Output isn't correct
3 Halted 0 ms 0 KB -