#include <bits/stdc++.h> using namespace std; template<typename T> using pair2 = pair<T, T>; #define int long long #define pii pair<int,int> #define pb push_back #define pf push_front #define mp make_pair #define all(x) (x).begin(),(x).end() #define fi first #define se second #define endl "\n" #define in(x) cin >> x #define ini(x) int x; in(x) #define instr(x) string x; in(x) #define inf 1e18 //https://codeforces.com/blog/entry/62393 struct custom_hash { static uint64_t splitmix64(uint64_t x) { // http://xorshift.di.unimi.it/splitmix64.c x += 0x9e3779b97f4a7c15; x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9; x = (x ^ (x >> 27)) * 0x94d049bb133111eb; return x ^ (x >> 31); } size_t operator()(uint64_t x) const { static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count(); return splitmix64(x + FIXED_RANDOM); } }; void dfs(int at, int prev, int color, int dist, int dist_max, vector<int> &colors, vector<unordered_map<int,int,custom_hash>> &graph) { colors[at] = color; for (auto &i : graph[at]) { if (dist + i.second <= dist_max) { if (i.first != prev) { dfs(i.first,at,color,dist+i.second,dist_max,colors,graph); } } } } signed main() { ios_base::sync_with_stdio(false); cin.tie(0); ini(n); ini(m); ini(q); vector<unordered_map<int,int,custom_hash>> graph(n); for (int i = 0; i < m; i++) { ini(a); ini(b); ini(d); a--; b--; graph[a][b] = d; graph[b][a] = d; } vector<int> colors(n); for (int i = 0; i < q; i++) { ini(type); if (type == 1) { ini(a); ini(b); ini(d); a--; b--; // dodajemy krawędź a->b o długości d graph[a][b] = d; graph[b][a] = d; } else if (type == 2) { ini(a); ini(b); a--; b--; // usuwamy krawędź {a,b} graph[a].erase(b); graph[b].erase(a); } else if (type == 3) { ini(v); ini(z); ini(k); v--; // przemalowujemy wszystkie wierzchołki odległe o maksymalnie z od v na kolor k dfs(v,-1,k,0,z,colors,graph); } else { ini(u); u--; // aktualny kolor u cout << colors[u] << endl; } } }
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 106 107 108 109 110 111 112 113 | #include <bits/stdc++.h> using namespace std; template<typename T> using pair2 = pair<T, T>; #define int long long #define pii pair<int,int> #define pb push_back #define pf push_front #define mp make_pair #define all(x) (x).begin(),(x).end() #define fi first #define se second #define endl "\n" #define in(x) cin >> x #define ini(x) int x; in(x) #define instr(x) string x; in(x) #define inf 1e18 //https://codeforces.com/blog/entry/62393 struct custom_hash { static uint64_t splitmix64(uint64_t x) { // http://xorshift.di.unimi.it/splitmix64.c x += 0x9e3779b97f4a7c15; x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9; x = (x ^ (x >> 27)) * 0x94d049bb133111eb; return x ^ (x >> 31); } size_t operator()(uint64_t x) const { static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count(); return splitmix64(x + FIXED_RANDOM); } }; void dfs(int at, int prev, int color, int dist, int dist_max, vector<int> &colors, vector<unordered_map<int,int,custom_hash>> &graph) { colors[at] = color; for (auto &i : graph[at]) { if (dist + i.second <= dist_max) { if (i.first != prev) { dfs(i.first,at,color,dist+i.second,dist_max,colors,graph); } } } } signed main() { ios_base::sync_with_stdio(false); cin.tie(0); ini(n); ini(m); ini(q); vector<unordered_map<int,int,custom_hash>> graph(n); for (int i = 0; i < m; i++) { ini(a); ini(b); ini(d); a--; b--; graph[a][b] = d; graph[b][a] = d; } vector<int> colors(n); for (int i = 0; i < q; i++) { ini(type); if (type == 1) { ini(a); ini(b); ini(d); a--; b--; // dodajemy krawędź a->b o długości d graph[a][b] = d; graph[b][a] = d; } else if (type == 2) { ini(a); ini(b); a--; b--; // usuwamy krawędź {a,b} graph[a].erase(b); graph[b].erase(a); } else if (type == 3) { ini(v); ini(z); ini(k); v--; // przemalowujemy wszystkie wierzchołki odległe o maksymalnie z od v na kolor k dfs(v,-1,k,0,z,colors,graph); } else { ini(u); u--; // aktualny kolor u cout << colors[u] << endl; } } } |