#329517 (C++14) No.421 しろくろチョコレート

提出ソース

結果

問題	No.421 しろくろチョコレート
ユーザー	Haar
提出日時	2019-03-27 23:08:30
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	TLE (最新) AC (最初)
実行時間	-
コード長	4,224 bytes
コンパイル時間	2,107 ms
コンパイル使用メモリ	186,960 KB
実行使用メモリ	101,760 KB
最終ジャッジ日時	2024-10-11 06:15:43
合計ジャッジ時間	15,821 ms
ジャッジサーバーID （参考情報）	judge2 / judge5

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	228 ms 66,944 KB
testcase_02	AC	104 ms 65,152 KB
testcase_03	AC	3 ms 5,248 KB
testcase_04	AC	50 ms 32,640 KB
testcase_05	AC	23 ms 8,704 KB
testcase_06	AC	3 ms 5,248 KB
testcase_07	AC	77 ms 20,096 KB
testcase_08	AC	117 ms 73,856 KB
testcase_09	AC	4 ms 5,248 KB
testcase_10	AC	12 ms 7,424 KB
testcase_11	AC	88 ms 25,216 KB
testcase_12	AC	2 ms 5,248 KB
testcase_13	AC	2 ms 5,248 KB
testcase_14	AC	2 ms 5,248 KB
testcase_15	AC	2 ms 5,248 KB
testcase_16	AC	505 ms 94,080 KB
testcase_17	AC	567 ms 101,760 KB
testcase_18	AC	583 ms 101,632 KB
testcase_19	AC	2 ms 5,248 KB
testcase_20	AC	233 ms 54,528 KB
testcase_21	AC	417 ms 69,632 KB
testcase_22	AC	102 ms 24,320 KB
testcase_23	AC	2 ms 5,248 KB
testcase_24	AC	2 ms 5,248 KB
testcase_25	AC	2 ms 5,248 KB
testcase_26	AC	2 ms 5,248 KB
testcase_27	AC	2 ms 5,248 KB
testcase_28	AC	139 ms 101,504 KB
testcase_29	AC	146 ms 101,504 KB
testcase_30	AC	229 ms 101,504 KB
testcase_31	AC	168 ms 101,760 KB
testcase_32	AC	138 ms 101,504 KB
testcase_33	AC	292 ms 101,504 KB
testcase_34	AC	125 ms 101,504 KB
testcase_35	AC	124 ms 101,504 KB
testcase_36	AC	123 ms 38,272 KB
testcase_37	AC	587 ms 79,616 KB
testcase_38	AC	772 ms 101,760 KB
testcase_39	AC	127 ms 54,528 KB
testcase_40	AC	249 ms 24,320 KB
testcase_41	AC	6 ms 6,144 KB
testcase_42	AC	9 ms 6,272 KB
testcase_43	AC	97 ms 47,104 KB
testcase_44	TLE	-
testcase_45	AC	47 ms 25,216 KB
testcase_46	AC	34 ms 16,384 KB
testcase_47	AC	219 ms 57,088 KB
testcase_48	AC	175 ms 64,512 KB
testcase_49	AC	110 ms 67,456 KB
testcase_50	AC	48 ms 30,976 KB
testcase_51	AC	325 ms 41,856 KB
testcase_52	AC	77 ms 23,680 KB
testcase_53	AC	35 ms 22,656 KB
testcase_54	AC	151 ms 76,416 KB
testcase_55	AC	7 ms 6,528 KB
testcase_56	AC	7 ms 6,272 KB
testcase_57	AC	33 ms 12,928 KB
testcase_58	AC	230 ms 56,704 KB
testcase_59	AC	34 ms 11,648 KB
testcase_60	AC	989 ms 101,632 KB
testcase_61	AC	675 ms 101,632 KB
testcase_62	AC	2 ms 5,248 KB
testcase_63	AC	2 ms 5,248 KB
testcase_64	AC	88 ms 101,632 KB

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

ソースコード

raw source code

#include <bits/stdc++.h>
#define FOR(v, a, b) for(int v = (a); v < (b); ++v)
#define FORE(v, a, b) for(int v = (a); v <= (b); ++v)
#define REP(v, n) FOR(v, 0, n)
#define REPE(v, n) FORE(v, 0, n)
#define REV(v, a, b) for(int v = (a); v >= (b); --v)
#define ALL(x) (x).begin(), (x).end()
#define ITR(it, c) for(auto it = (c).begin(); it != (c).end(); ++it)
#define RITR(it, c) for(auto it = (c).rbegin(); it != (c).rend(); ++it)
#define EXIST(c,x) ((c).find(x) != (c).end())
#define LLI long long int
#define fst first
#define snd second

#ifdef DEBUG
#include <misc/C++/Debug.cpp>
#else
#define dump(x) ((void)0)
#endif

using namespace std;

#define gcd __gcd
template <class T> constexpr T lcm(T m, T n){return m/gcd(m,n)*n;}

template <typename I> void join(ostream &ost, I s, I t, string d=" "){for(auto i=s; i!=t; ++i){if(i!=s)ost<<d; ost<<*i;}ost<<endl;}
template <typename T> istream& operator>>(istream &is, vector<T> &v){for(auto &a : v) is >> a; return is;}
template <typename T, typename U> istream& operator>>(istream &is, pair<T,U> &p){is >> p.first >> p.second; return is;}

template <typename T, typename U> T& chmin(T &a, const U &b){return a = (a<=b?a:b);}
template <typename T, typename U> T& chmax(T &a, const U &b){return a = (a>=b?a:b);}
template <typename T, size_t N, typename U> void fill_array(T (&a)[N], const U &v){fill((U*)a, (U*)(a+N), v);}

template <typename T, T INF> class Dinic{
private:
  vector<vector<pair<int,T>>> graph;
  int size, s, t;
  vector<vector<T>> cap;
  vector<int> level;
  
  bool buildLevel(){
    fill(ALL(level), 0);
    level[s] = 1;
    deque<int> deq = {s};
    while(!deq.empty()){
      int cur = deq.front(); deq.pop_front();
      REP(i,size)
	if(level[i]==0 && cap[cur][i]>0){
	  level[i] = level[cur] + 1;
	  deq.push_back(i);
	}
    }
    return level[t] != 0;
  }
  void dfs(vector<int> &path, T &flow){
    if(path.empty()) return;
    int cur = path.back();
    if(cur == t){
      T f = INF;
      FOR(i,1,path.size()) f = min(f, cap[path[i-1]][path[i]]);
      FOR(i,1,path.size()){
	cap[path[i-1]][path[i]] -= f;
	cap[path[i]][path[i-1]] += f;
      }
      flow += f;
    }else{
      REP(i,size){
	if(cap[cur][i]>0 && level[i]>level[cur]){
	  path.push_back(i);
	  dfs(path, flow);
	  path.pop_back();
	}
      }
    }
  }
  T augment(){
    T f = 0;
    vector<int> path = {s};
    dfs(path, f);
    return f;
  }
  T loop(){
    T f = 0;
    while(buildLevel()) f += augment();
    return f;
  }

public:
  Dinic(vector<vector<pair<int,T>>> &_graph): graph(_graph), size(graph.size()) {}
  Dinic(int size): graph(size), size(size){}

  void add_edge(int from, int to, const T &cap){
    graph[from].push_back({to, cap});
  }
  
  T max_flow(int _s, int _t){
    cap = vector<vector<T>>(size, vector<T>(size, 0));
    level = vector<int>(size, 0);

    REP(i,size)
      for(auto &p : graph[i]){
	int j = p.first;
	T d = p.second;
	cap[i][j] += d;
      }

    s = _s;
    t = _t;

    return loop();
  }
};

class BipartiteMatching{
public:
  int x, y;
  Dinic<int,INT_MAX> mflow;
  int s, t;
  BipartiteMatching(int x, int y): x(x), y(y), mflow(x+y+2), s(x+y), t(s+1){
    REP(i,x) mflow.add_edge(s,i,1);
    REP(i,y) mflow.add_edge(x+i,t,1);
  }

  void add_edge(int i, int j){
    mflow.add_edge(i,x+j,1);
  }

  int matching(){
    return mflow.max_flow(s,t);
  }
};

const int dir4[4][2] = {{1,0},{-1,0},{0,1},{0,-1}};

int main(){
  cin.tie(0);
  ios::sync_with_stdio(false);

  int N,M;
  while(cin >> N >> M){
    vector<string> S(N); cin >> S;
    int c = 0;
    REP(i,N) REP(j,M) if(S[i][j] != '.') ++c;

    BipartiteMatching bm(N*M,N*M);

    vector<vector<int>> w(N, vector<int>(M,-1)), b(N, vector<int>(M,-1));
    int cw=0, cb=0;
    REP(i,N) REP(j,M){
      if(S[i][j] == 'w') w[i][j] = cw++;
      if(S[i][j] == 'b') b[i][j] = cb++;
    }

    REP(i,N){
      REP(j,M){
        for(auto &d : dir4){
	  int y=i+d[0], x=j+d[1];
	  if(y<0 or y>=N or x<0 or x>=M) continue;

	  if(S[i][j] == 'w' and S[y][x] == 'b') bm.add_edge(w[i][j], b[y][x]);
	}
      }
    }
    
    int mat = bm.matching();
    int pr = min(cw-mat,cb-mat);
    int ans = mat*100 + pr*10 + (cw-mat-pr) + (cb-mat-pr);

    cout << ans << endl;
  }
  
  return 0;
}

yukicoder

結果

テストケース

ソースコード