Submission #1195367

---

Source Code Expand

#include <iostream>
#include <string>
#include <vector>
#include <utility>
#include <map>
#include <algorithm>

using namespace std;

void trace(
  const vector<vector<int>>& vertices,
  vector<int>& colors,
  int old_v,
  int v,
  int d,
  int c
) {
  // cerr << "trace: " << (old_v+1) << ", " << (v+1) << ", " << d << ", " << c << endl;
  colors[v] = c;
  if (d <= 0) return;
  for (auto next_v: vertices[v]) {
    if (next_v == old_v) continue;
    trace(vertices, colors, v, next_v, d - 1, c);
  }
}

void paint(
  const vector<vector<int>>& vertices,
  vector<int>& colors,
  int v,
  int d,
  int c
) {
  // cerr << "paint: " << (v+1) << ", " << d << ", " << c << endl;
  trace(vertices, colors, v, v, d, c);
}

int main()
{
  int N, M;
  cin >> N >> M;
  vector<vector<int>> vertices(N);
  vector<int> colors(N, 0);
  for (int i = 0; i < M; ++i) {
    int a, b;
    cin >> a >> b;
    --a; --b;
    vertices[a].push_back(b);
    vertices[b].push_back(a);
  }

  int Q;
  cin >> Q;
  for (int i = 0; i < Q; ++i) {
    int v, d, c;
    cin >> v >> d >> c;
    --v;
    paint(vertices, colors, v, d, c);
  }

  for (auto c: colors) cout << c << endl;
  return 0;
}

Submission Info

Judge Result