#61145 (C++11) No.307 最近色塗る問題多くない？

提出ソース

結果

問題	No.307 最近色塗る問題多くない？
ユーザー	is_eri23
提出日時	2015-11-28 17:07:05
言語	C++11 (gcc 11.4.0)
結果	MLE
実行時間	-
コード長	5,304 bytes
コンパイル時間	1,958 ms
コンパイル使用メモリ	187,904 KB
実行使用メモリ	469,664 KB
最終ジャッジ日時	2024-09-14 04:54:55
合計ジャッジ時間	11,078 ms
ジャッジサーバーID （参考情報）	judge2 / judge6

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 13,884 KB
testcase_01	AC	1 ms 6,940 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	3 ms 6,944 KB
testcase_05	AC	1 ms 6,940 KB
testcase_06	AC	2 ms 6,940 KB
testcase_07	AC	8 ms 6,940 KB
testcase_08	AC	37 ms 6,940 KB
testcase_09	AC	15 ms 6,940 KB
testcase_10	AC	19 ms 6,944 KB
testcase_11	AC	111 ms 15,360 KB
testcase_12	AC	2 ms 6,940 KB
testcase_13	AC	14 ms 6,940 KB
testcase_14	AC	4 ms 6,940 KB
testcase_15	AC	4 ms 6,940 KB
testcase_16	AC	5 ms 6,944 KB
testcase_17	AC	19 ms 6,940 KB
testcase_18	AC	51 ms 10,112 KB
testcase_19	AC	61 ms 11,520 KB
testcase_20	AC	4 ms 6,940 KB
testcase_21	AC	104 ms 17,536 KB
testcase_22	AC	112 ms 16,896 KB
testcase_23	AC	123 ms 17,280 KB
testcase_24	AC	138 ms 17,664 KB
testcase_25	AC	172 ms 17,152 KB
testcase_26	AC	139 ms 17,792 KB
testcase_27	AC	1,862 ms 197,760 KB
testcase_28	MLE	-
testcase_29	--	-
testcase_30	--	-
testcase_31	--	-
testcase_32	--	-
testcase_33	--	-
testcase_34	--	-
testcase_35	--	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
#define EPS 1e-9
#define INF 1070000000LL
#define MOD 1000000007LL
#define fir first
#define foreach(it,X) for(auto it=(X).begin();it!=(X).end();it++)
#define numa(x,a) for(auto x: a)
#define ite iterator
#define mp make_pair
#define rep(i,n) rep2(i,0,n)
#define rep2(i,m,n) for(int i=m;i<(n);i++)
#define pb push_back
#define pf push_front
#define sec second
#define sz(x) ((int)(x).size())
#define ALL( c ) (c).begin(), (c).end()
#define gcd(a,b) __gcd(a,b)
#define mem(x,n) memset(x,n,sizeof(x))
#define endl "\n"
using namespace std;
template <int POS, class TUPLE> void deploy(std::ostream &os, const TUPLE &tuple){}
template <int POS, class TUPLE, class H, class ...Ts> void deploy(std::ostream &os, const TUPLE &t){ os << (POS == 0 ? "" : ", ") << get<POS>(t); deploy<POS + 1, TUPLE, Ts...>(os, t); }
template <class T,class U> std::ostream& operator<<(std::ostream &os, std::pair<T,U> &p){ os << "(" << p.first <<", " << p.second <<")";return os; }
template <class T> std::ostream& operator<<(std::ostream &os, std::vector<T> &v){ int remain = v.size(); os << "{"; for(auto e: v) os << e << (--remain == 0 ? "}" : ", "); return os; }
template <class T> std::ostream& operator<<(std::ostream &os, std::set<T> &v){ int remain = v.size(); os << "{"; for(auto e: v) os << e << (--remain == 0 ? "}" : ", "); return os; }
template <class T, class K> std::ostream& operator<<(std::ostream &os, std::map<T, K> &mp){ int remain = mp.size(); os << "{"; for(auto e: mp) os << "(" << e.first << " -> " << e.second << ")" << (--remain == 0 ? "}" : ", "); return os; }
#define DEBUG1(var0) { std::cerr << (#var0) << "=" << (var0) << endl; }
#define DEBUG2(var0, var1) { std::cerr << (#var0) << "=" << (var0) << ", ";DEBUG1(var1); }
#define DEBUG3(var0, var1, var2) { std::cerr << (#var0) << "=" << (var0) << ", ";DEBUG2(var1,var2); }
#define DEBUG4(var0, var1, var2, var3) { std::cerr << (#var0) << "=" << (var0) << ", ";DEBUG3(var1,var2,var3); }
using ll = long long;
int H,W;
char board[300][300];

struct A{
  vector <pair <int,int> > bounds;
  set <pair <int,int> > adj;
  pair <int,int> root;
  int color;
  A(){};
  A(int color_, const vector <pair <int,int> > bounds_, const pair <int,int> root_){
    color = color_;
    bounds = bounds_;
    root = root_;
  }
  
};

map <pair <int,int>, A> areas;
bool used[300][300];
pair <int,int> root[300][300];
vector <pair <int,int> > area;
int dx[] = {-1,0,0,1};
int dy[] = {0,1,-1,0};
inline bool OK(int x , int y){
  return 0 <= x && x < H && 0 <= y && y < W;
}
inline void dfs(int i, int j, pair <int,int> myroot){
  if (used[i][j]) {
    return;
  }
  used[i][j] = true;
  root[i][j] = myroot;
  area.pb(mp(i,j));
  rep(k,4){
    int nx = i + dx[k], ny = j + dy[k];
    if (OK(nx,ny) && board[i][j] == board[nx][ny]) {
      dfs(nx,ny,myroot);
    }
  }
}
inline vector <pair <int,int> > build_bound(){
  vector <pair <int,int > > bounds;
  for (auto a : area){
    rep(i,4){
      int ax = a.fir + dx[i];
      int ay = a.sec + dy[i];
      if (OK(ax, ay)) {
        if (!used[ax][ay] || root[ax][ay] != root[a.fir][a.sec]) {
          bounds.pb(a);
          break;
        }
      }else{
        bounds.pb(a);
        break;
      }
    }
  }
  return bounds;
}
inline void build_adj(){
  for (auto &a : areas) {
    for (auto &b : a.sec.bounds){
      rep(i,4){
        int nx = b.fir + dx[i];
        int ny = b.sec + dy[i];
        if (OK(nx,ny) && root[nx][ny] != root[b.fir][b.sec]) {
          a.sec.adj.insert(root[nx][ny]);
        }
      }
    }
  }
}
inline void build(){
  rep(i,H){
    rep(j,W){
      if (!used[i][j]) {
        dfs(i,j,mp(i,j));
        areas[mp(i,j)] = A(board[i][j] - '0', build_bound(), mp(i,j));
        area.clear();
      }
    }
  }
  build_adj();
}
inline pair <int,int> parent(const pair <int,int> &a){
  if (areas[a].root == a) {
    return a;
  }else{
    return areas[a].root = parent(areas[a].root);
  }
}
void print(){
  rep(i,H){
    rep(j,W){
      if (j) {
        cout << ' ';
      }
      auto myroot = parent(root[i][j]);
      char c = '0' + areas[myroot].color;
      cout << c;
    }
    cout << endl;
  }
   rep(i,H){
    rep(j,W){
      DEBUG3(i,j,root[i][j]);
      auto myroot = parent(root[i][j]);
      DEBUG3(i,j, myroot);
    }
  }
}

void paint(int R,int C, int X){
  auto myroot = parent(root[R][C]);
  auto& cur = areas[myroot];
  if (cur.color == X) {
    return;
  }
  
  set <pair <int,int> > new_adj;
  for (auto &a : cur.adj){
    //merge
    auto adj_root = parent(a);
    for (auto &b : areas[adj_root].adj) {
      new_adj.insert(parent(b));
    }
    areas[adj_root].root = myroot;
  }
  for (auto it = new_adj.begin(); it != new_adj.end();){
    if (parent(*it) == myroot) {
      new_adj.erase(it++);
    }else{
      it++;
    }
  }
  cur.adj = new_adj;
  cur.color = X;
  //print();
}

int main()
{
  cin.tie(0);
  ios_base::sync_with_stdio(0);
  cin >> H >> W;
  rep(i,H){
    rep(j,W){
      cin >> board[i][j];
    }
  }
  build();
  int Q;
  cin >> Q;
  rep(i,Q){
    int R,C,X;
    cin >> R >> C >> X;
    R--;C--;
    paint(R,C,X);
  }
 
  rep(i,H){
    rep(j,W){
      if (j) cout << ' ';
      auto myroot = parent(areas[root[i][j]].root);
      char c = areas[myroot].color + '0';
      cout << c;
    }
    cout << endl;
  }
  return 0;
}

yukicoder

結果

テストケース

ソースコード