#include<bits/stdc++.h>
#define FOR(i,a,b) for(int i=a;i<b;++i)
#define FORD(i,a,b) for(int i=a;i>=b;--i)
#define PB push_back
#define EB emplace_back
#define FI first
#define SE second
#define umap unordered_map
#define uset unordered_set
#define vi vector<int>
#define vvi vector<vi>
#define vll vector<ll>
#define vvll vector<vll>
#define vpii vector<pii>
#define pii pair<int, int>
#define pll pair<ll, ll>
#define ALL(X) (X).begin(),(X).end()
#ifndef DEBUG
#define endl (char)10
#endif
using namespace std;
using ll = long long;
using ld = long double;
template <class T>
istream& operator>> (istream& is, vector<T>& vec){
FOR(i,0,vec.size()) is >> vec[i];
return is;
}
template <class T>
ostream& operator<< (ostream& os, vector<T>& vec){
for(auto& t : vec) os << t << " ";
return os;
}
template<class T, class U>
ostream& operator<< (ostream& os, const pair<T, U>& p){
os << p.FI << " " << p.SE;
return os;
}
template<class T, class U>
istream& operator>> (istream& is, pair<T, U>& p){
is >> p.FI >> p.SE;
return is;
}
struct query {
int t;
int a;
ll b;
int c;
};
int main () {
ios_base::sync_with_stdio(false);
cin.tie(0);
int n, m, qqq;
cin >> n >> m >> qqq;
vector<map<int, int>> V(n);
vector<query> Q(qqq);
FOR(i,0,m){
int a, b, d;
cin >> a >> b >> d;
a--; b--;
V[a][b] = d;
V[b][a] = d;
}
FOR(i,0,qqq){
cin >> Q[i].t >> Q[i].a;
Q[i].a--;
if (Q[i].t == 1) {
cin >> Q[i].b >> Q[i].c;
Q[i].b--;
} else if (Q[i].t == 2) {
cin >> Q[i].b;
Q[i].b--;
} else if (Q[i].t == 3) cin >> Q[i].b >> Q[i].c;
}
vi color(n, 0);
vll D(n);
for(auto& q : Q){
if (q.t == 4){
cout << color[q.a] << endl;
} else if (q.t == 1){
V[q.a][q.b] = q.c;
V[q.b][q.a] = q.c;
} else if (q.t == 2) {
V[q.a].erase(q.b);
V[q.b].erase(q.a);
} else {
FOR(i,0,n) D[i] = q.b + 1;
priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<pair<ll, int>>> K;
D[q.a] = 0;
K.emplace(0, q.a);
while(!K.empty()){
auto c = K.top();
//cout << "Sciaglem " << c << endl;
K.pop();
if (D[c.SE] != c.FI) continue;
//cout << "RECOLOR " << c.SE << " " << q.c << endl;
color[c.SE] = q.c;
for(const auto& s : V[c.SE]){
ll nd = c.FI + s.SE;
if (nd < D[s.FI]) {
K.emplace(nd, s.FI);
D[s.FI] = nd;
}
}
}
}
}
}
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 | #include<bits/stdc++.h> #define FOR(i,a,b) for(int i=a;i<b;++i) #define FORD(i,a,b) for(int i=a;i>=b;--i) #define PB push_back #define EB emplace_back #define FI first #define SE second #define umap unordered_map #define uset unordered_set #define vi vector<int> #define vvi vector<vi> #define vll vector<ll> #define vvll vector<vll> #define vpii vector<pii> #define pii pair<int, int> #define pll pair<ll, ll> #define ALL(X) (X).begin(),(X).end() #ifndef DEBUG #define endl (char)10 #endif using namespace std; using ll = long long; using ld = long double; template <class T> istream& operator>> (istream& is, vector<T>& vec){ FOR(i,0,vec.size()) is >> vec[i]; return is; } template <class T> ostream& operator<< (ostream& os, vector<T>& vec){ for(auto& t : vec) os << t << " "; return os; } template<class T, class U> ostream& operator<< (ostream& os, const pair<T, U>& p){ os << p.FI << " " << p.SE; return os; } template<class T, class U> istream& operator>> (istream& is, pair<T, U>& p){ is >> p.FI >> p.SE; return is; } struct query { int t; int a; ll b; int c; }; int main () { ios_base::sync_with_stdio(false); cin.tie(0); int n, m, qqq; cin >> n >> m >> qqq; vector<map<int, int>> V(n); vector<query> Q(qqq); FOR(i,0,m){ int a, b, d; cin >> a >> b >> d; a--; b--; V[a][b] = d; V[b][a] = d; } FOR(i,0,qqq){ cin >> Q[i].t >> Q[i].a; Q[i].a--; if (Q[i].t == 1) { cin >> Q[i].b >> Q[i].c; Q[i].b--; } else if (Q[i].t == 2) { cin >> Q[i].b; Q[i].b--; } else if (Q[i].t == 3) cin >> Q[i].b >> Q[i].c; } vi color(n, 0); vll D(n); for(auto& q : Q){ if (q.t == 4){ cout << color[q.a] << endl; } else if (q.t == 1){ V[q.a][q.b] = q.c; V[q.b][q.a] = q.c; } else if (q.t == 2) { V[q.a].erase(q.b); V[q.b].erase(q.a); } else { FOR(i,0,n) D[i] = q.b + 1; priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<pair<ll, int>>> K; D[q.a] = 0; K.emplace(0, q.a); while(!K.empty()){ auto c = K.top(); //cout << "Sciaglem " << c << endl; K.pop(); if (D[c.SE] != c.FI) continue; //cout << "RECOLOR " << c.SE << " " << q.c << endl; color[c.SE] = q.c; for(const auto& s : V[c.SE]){ ll nd = c.FI + s.SE; if (nd < D[s.FI]) { K.emplace(nd, s.FI); D[s.FI] = nd; } } } } } } |
English