-
Notifications
You must be signed in to change notification settings - Fork 11
/
Copy path1778.cpp
105 lines (105 loc) · 3.01 KB
/
1778.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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
// Ivan Carvalho
// Solution to https://www.beecrowd.com.br/judge/problems/view/1778
#include <cstdio>
#define MAXN 1001
typedef long long ll;
const ll INF = 1e15;
ll n, m, f, p, q, TC;
ll monstro[10 * MAXN], hpmonstro[10 * MAXN];
ll adj[MAXN][MAXN], grau[MAXN];
ll fila1[MAXN * (MAXN + 10)], fila2[MAXN * (MAXN + 10)], processado[MAXN], ini,
fim, origem, capacidade;
ll custo[MAXN], matriz[MAXN][MAXN], distancia[MAXN], dano;
void bfs() {
ini = fim = 1;
fila1[ini] = origem;
fila2[ini] = 0;
for (ll i = 1; i <= n; i++) {
processado[i] = 0;
}
processado[f] = 1;
while (ini <= fim && fila2[ini] <= capacidade) {
// printf("bfs\n");
ll vertice = fila1[ini];
ll percorrido = fila2[ini++];
if (processado[vertice]) continue;
processado[vertice] = 1;
custo[vertice] += dano;
for (ll i = 0; i < grau[vertice]; i++) {
ll u = adj[vertice][i];
if (processado[u]) continue;
fila1[++fim] = u;
fila2[fim] = percorrido + 1;
}
}
}
void Dijkstra() {
for (ll i = 1; i <= n; i++) {
distancia[i] = matriz[origem][i];
processado[i] = 0;
}
processado[origem] = 1;
distancia[origem] = 0;
while (true) {
ll davez = -1;
ll menor = INF;
for (ll i = 1; i <= n; i++) {
if (!processado[i] && distancia[i] < menor) {
davez = i;
menor = distancia[i];
}
}
if (davez == -1) break;
processado[davez] = 1;
for (ll i = 1; i <= n; i++) {
if (distancia[davez] + matriz[davez][i] < distancia[i]) {
distancia[i] = distancia[davez] + matriz[davez][i];
}
}
}
}
int main() {
scanf("%lld", &TC);
for (ll tc = 1; tc <= TC; tc++) {
scanf("%lld %lld %lld", &n, &m, &f);
for (ll i = 1; i <= n; i++) {
grau[i] = 0;
custo[i] = 0;
for (ll j = 1; j <= n; j++) {
matriz[i][j] = INF;
}
}
while (m--) {
ll u, v;
scanf("%lld %lld", &u, &v);
adj[u][grau[u]++] = v;
adj[v][grau[v]++] = u;
}
scanf("%lld", &p);
while (p--) {
scanf("%lld %lld %lld", &origem, &dano, &capacidade);
bfs();
}
scanf("%lld", &q);
for (ll i = 1; i <= q; i++) {
scanf("%lld %lld", &monstro[i], &hpmonstro[i]);
}
for (ll i = 1; i <= n; i++) {
for (ll j = 0; j < grau[i]; j++) {
matriz[i][adj[i][j]] = custo[adj[i][j]];
}
matriz[i][i] = 0;
}
origem = f;
Dijkstra();
ll resp = 0;
for (ll i = 1; i <= q; i++) {
// printf("%lld %lld\n",(ll)hpmonstro[i],distancia[monstro[i]]);
if ((ll)hpmonstro[i] > distancia[monstro[i]]) {
resp++;
}
}
printf("Caso #%lld: %lld\n", tc, resp);
}
return 0;
}