結果

問題 No.307 最近色塗る問題多くない？
ユーザー imulanimulan
提出日時 2016-10-02 23:32:50
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
MLE  
実行時間 -
コード長 3,950 bytes
コンパイル時間 2,191 ms
コンパイル使用メモリ 192,536 KB
実行使用メモリ 813,952 KB
最終ジャッジ日時 2024-05-01 10:48:28
合計ジャッジ時間 8,594 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
6,816 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 3 ms
6,940 KB
testcase_04 AC 4 ms
6,944 KB
testcase_05 AC 3 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 16 ms
11,648 KB
testcase_08 AC 213 ms
120,320 KB
testcase_09 AC 111 ms
65,152 KB
testcase_10 AC 34 ms
22,912 KB
testcase_11 AC 205 ms
123,520 KB
testcase_12 AC 3 ms
6,944 KB
testcase_13 AC 15 ms
6,940 KB
testcase_14 AC 7 ms
7,936 KB
testcase_15 AC 13 ms
10,752 KB
testcase_16 AC 12 ms
11,392 KB
testcase_17 AC 152 ms
88,832 KB
testcase_18 MLE -
testcase_19 MLE -
testcase_20 AC 7 ms
7,168 KB
testcase_21 MLE -
testcase_22 MLE -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
#define rep(i,n) for(int (i)=0;(i)<(int)(n);++(i))
#define each(itr,c) for(__typeof(c.begin()) itr=c.begin(); itr!=c.end(); ++itr)
#define all(x) (x).begin(),(x).end()
#define pb push_back
#define fi first
#define se second

typedef pair<int,int> pi;

struct UF{
    int n;
    //正だったらその頂点の親,負だったら根で連結成分の個数
    vector<int> d;
    UF() {}
    UF(int N):n(N), d(N,-1){}
    int root(int v){
        if(d[v]<0) return v;
        return d[v]=root(d[v]);
    }
    bool unite(int X,int Y){
        X=root(X); Y=root(Y);
        if(X==Y) return false;
        if(size(X) < size(Y)) swap(X,Y);
        d[X]+=d[Y];
        d[Y]=X;
        return true;
    }
    int size(int v){ return -d[root(v)]; }
};

int h,w;
int a[200][200];

// connected component
int cc[200][200];
int dx[4]={1,-1,0,0}, dy[4]={0,0,1,-1};

// Graph
set<int> G[40000];
int col[40000];

void merge_set(int x, int y)
{
    if(x==y) return;
    if(G[x].size()<G[y].size()) swap(x,y);
    G[x].insert(G[y].begin(), G[y].end());
    G[y].clear();
    G[y]=G[x];
}

int main()
{
    scanf(" %d %d", &h, &w);
    rep(i,h)rep(j,w) scanf(" %d", &a[i][j]);

    // BFS
    #define IN(x,y) (0<=x && x<w && 0<=y && y<h)
    memset(cc,-1,sizeof(cc));
    int cc_num=0;
    rep(y,h)rep(x,w)
    {
        if(cc[y][x]>=0) continue;

        queue<pi> que;
        que.push(pi(y,x));
        cc[y][x]=cc_num;
        while(!que.empty())
        {
            pi now=que.front();
            que.pop();
            rep(i,4)
            {
                int nx=now.se+dx[i], ny=now.fi+dy[i];
                if(IN(nx,ny) && a[ny][nx]==a[now.fi][now.se] && cc[ny][nx]==-1)
                {
                    cc[ny][nx]=cc_num;
                    que.push(pi(ny,nx));
                }
            }
        }

        ++cc_num;
    }

    // グラフ構成
    rep(y,h)rep(x,w)
    {
        col[cc[y][x]] = a[y][x];
        rep(i,4)
        {
            int nx=x+dx[i], ny=y+dy[i];
            if(IN(nx,ny) && cc[y][x]!=cc[ny][nx])
            {
                int p=cc[y][x], q=cc[ny][nx];
                G[p].insert(q);
                G[q].insert(p);
            }
        }
    }

    // rep(i,h)
    // {
    //     rep(j,w) printf("%3d", cc[i][j]);
    //     printf("\n");
    // }
    // rep(i,cc_num)
    // {
    //     printf(" %d(col=%d) :", i,col[i]);
    //     for(const int &j:G[i]) printf(" %d", j);
    //     printf("\n");
    // }

    UF uf(cc_num);

    // クエリ処理
    int Q;
    scanf(" %d", &Q);
    while(Q--)
    {
        int r,c,x;
        scanf(" %d %d %d", &r, &c, &x);
        --r;
        --c;

        if(x == col[ uf.root(cc[r][c]) ]) continue;

        vector<int> tmp(G[uf.root(cc[r][c])].begin(), G[uf.root(cc[r][c])].end());

        for(auto &i:tmp) i=uf.root(i);
        sort(all(tmp));
        tmp.erase(unique(all(tmp)), tmp.end());

        G[uf.root(cc[r][c])].clear();
        G[uf.root(cc[r][c])].insert(all(tmp));

        set<int> tmpG=G[uf.root(cc[r][c])];
        for(const int &nx:tmpG)
        {
            if(col[uf.root(nx)]==x)
            {
                // printf("mergeing: %d and %d\n", uf.root(cc[r][c]), uf.root(nx));
                merge_set(uf.root(cc[r][c]), uf.root(nx));
                uf.unite(uf.root(cc[r][c]),uf.root(nx));
            }
        }

        col[uf.root(cc[r][c])] = x;

        // printf(" process: %d\n",Q);
        // rep(i,cc_num)
        // {
        //     printf(" %d(col=%d) :", i,col[i]);
        //     for(const int &j:G[i]) printf(" %d", j);
        //     printf("\n");
        // }
    }

    // rep(y,h)rep(x,w) printf("%d,%d -> root: %d\n", y,x,uf.root(cc[y][x]) );

    // 出力
    rep(y,h)
    {
        rep(x,w)
        {
            if(x) printf(" ");
            printf("%d", col[uf.root(cc[y][x])]);
        }
        printf("\n");
    }
    return 0;
}
0