結果

問題 No.307 最近色塗る問題多くない？
ユーザー pekempeypekempey
提出日時 2015-11-28 00:08:01
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 3,011 ms / 4,000 ms
コード長 2,489 bytes
コンパイル時間 1,503 ms
コンパイル使用メモリ 170,488 KB
実行使用メモリ 13,184 KB
最終ジャッジ日時 2024-11-21 21:56:11
合計ジャッジ時間 9,083 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 36
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:69:36: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   69 |         rep (i, h) rep (j, w) scanf("%d", &A[i][j]);
      |                               ~~~~~^~~~~~~~~~~~~~~~
main.cpp:91:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   91 |                 scanf("%d %d %d", &r, &c, &x);
      |                 ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>
#define GET_MACRO(a, b, c, NAME, ...) NAME
#define rep(...) GET_MACRO(__VA_ARGS__, rep3, rep2)(__VA_ARGS__)
#define rep2(i, a) rep3 (i, 0, a)
#define rep3(i, a, b) for (int i = (a); i < (b); i++)
#define repr(...) GET_MACRO(__VA_ARGS__, repr3, repr2)(__VA_ARGS__)
#define repr2(i, a) repr3 (i, 0, a)
#define repr3(i, a, b) for (int i = (b) - 1; i >= (a); i--)
#define chmin(a, b) ((b) < a && (a = (b), true))
#define chmax(a, b) (a < (b) && (a = (b), true))
using namespace std;
typedef long long ll;

struct UF {
    vector<int> parent, color;
    vector<set<int>> g;
    UF(int n) : parent(n), color(n), g(n) {
        rep (i, n) parent[i] = i;
    }
    int root(int x) {
        if (x == parent[x]) return x;
        else return parent[x] = root(parent[x]);
    }
    void merge(int x, int y) {
        x = root(x); y = root(y);
        if (x == y) return;
        if (g[x].size() < g[y].size()) swap(x, y);
        parent[y] = x;
        g[x].insert(g[y].begin(), g[y].end());
        g[y].clear();

        auto it = g[x].begin();
        while (it != g[x].end()) {
            if (root(*it) == x) {
                g[x].erase(it++);
            } else ++it;
        }
    }
    bool same(int x, int y) {
        return root(x) == root(y);
    }
    void add(int u, int v) {
        u = root(u); v = root(v);
        if (u == v) return;
        g[u].insert(v);
        g[v].insert(u);
    }
    void update(int x, int c) {
        x = root(x);
        if (color[x] == c) return;
        color[x] = c;
        vector<int> a(g[x].begin(), g[x].end());
        for (int y : a) if (getcolor(y) == c) {
            merge(x, y);
        }
    }
    int getcolor(int x) {
        x = root(x);
        return color[x];
    }
};
int g[200][200];
int A[200][200];

int main() {
	int h, w;
	cin >> h >> w;
	UF uf(h * w);
	rep (i, h) rep (j, w) scanf("%d", &A[i][j]);
	rep (i, h) rep (j, w) uf.color[i * w + j] = A[i][j];
	rep (i, h) rep (j, w) {
		int dy[] = {1, 0};
		int dx[] = {0, 1};
		rep (k, 2) {
			int ny = i + dy[k];
			int nx = j + dx[k];
			if (ny < h && nx < w) {
				if (A[ny][nx] == A[i][j]) {
					uf.merge(ny * w + nx, i * w + j);
				} else {
					uf.add(ny * w + nx, i * w + j);
				}
			}
		}
	}

	int q;
	cin >> q;
	rep (i, q) {
		int r, c, x;
		scanf("%d %d %d", &r, &c, &x);
		r--; c--;
		uf.update(r * w + c, x);
	}

	rep (i, h) {
		rep (j, w) {
			if (j) printf(" ");
			printf("%d", uf.getcolor(i * w + j));
		}
		printf("\n");
	}
	return 0;
}
0