-
Notifications
You must be signed in to change notification settings - Fork 59
/
Copy patharb2.cpp
66 lines (56 loc) · 1.36 KB
/
arb2.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# include <cstdio>
# include <algorithm>
# include <vector>
# define inf 999999999
# define NR 100005
using namespace std;
struct elem
{
int y, dist, cost;
}E;
vector <elem> v[NR];
int i,j,n,m,x,y,maxx;
int H[NR], C[NR];
long long COST;
void DFS (int k, int tata)
{
if (v[k].size()==1) C[k]=inf;
for (int i=0; i<v[k].size(); ++i)
if (v[k][i].y!=tata)
{
DFS (v[k][i].y, k);
H[k]=max(H[k], H[v[k][i].y] + v[k][i].dist);
C[v[k][i].y]=min(C[v[k][i].y], v[k][i].cost);
C[k]=C[k] + C[v[k][i].y];
}
}
void calc (int k, int tata)
{
for (int i=0; i<v[k].size(); ++i)
if (v[k][i].y!=tata)
{
if (v[k][i].dist + H[v[k][i].y] < H[k]) // trebuie sa-l crestem
{
COST=COST + 1LL*C[v[k][i].y] * (H[k] - H[v[k][i].y] - v[k][i].dist);
}
calc (v[k][i].y, k);
}
}
int main ()
{
freopen ("arb2.in", "r", stdin);
freopen ("arb2.out", "w", stdout);
scanf ("%d", &n);
for (i=1; i<n; ++i)
{
scanf ("%d%d%d%d", &x, &y, &E.dist, &E.cost);
E.y=y; v[x].push_back(E);
E.y=x; v[y].push_back(E);
}
// C[i] - costul minim ca sa ridicam tot subarborele
// lui i cu o unitate
DFS (1, 0); //distante + costuri
calc (1, 0);
printf ("%lld\n", COST);
return 0;
}