結果

問題 No.861 ケーキカット
コンテスト
ユーザー ふっぴー
提出日時 2019-08-09 22:52:39
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 28 ms / 1,000 ms
コード長 4,720 bytes
コンパイル時間 2,065 ms
コンパイル使用メモリ 182,388 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-07 19:02:35
合計ジャッジ時間 2,984 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <unistd.h>

using namespace std;

#define DEBUG(x) cerr<<#x<<": "<<x<<endl;
#define DEBUG_VEC(v) cerr<<#v<<":";for(int i=0;i<v.size();i++) cerr<<" "<<v[i]; cerr<<endl
#define DEBUG_MAT(v) cerr<<#v<<endl;for(int i=0;i<v.size();i++){for(int j=0;j<v[i].size();j++) {cerr<<v[i][j]<<" ";}cerr<<endl;}


typedef long long ll;
#define vi vector<int>
#define vl vector<ll>
#define vii vector< vector<int> >
#define vll vector< vector<ll> >
#define vs vector<string>
#define pii pair<int,int>
#define pis pair<int,string>
#define psi pair<string,int>
#define pll pair<ll,ll>
template<class S, class T> pair<S, T> operator+(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first + t.first, s.second + t.second); }
template<class S, class T> pair<S, T> operator-(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first - t.first, s.second - t.second); }
template<class S, class T> ostream& operator<<(ostream& os, pair<S, T> p) { os << "(" << p.first << ", " << p.second << ")"; return os; }
#define X first
#define Y second
#define rep(i,n) for(int i=0;i<(n);i++)
#define rep1(i,n) for(int i=1;i<=(n);i++)
#define rrep(i,n) for(int i=(n)-1;i>=0;i--)
#define rrep1(i,n) for(int i=(n);i>0;i--)
#define REP(i,a,b) for(int i=a;i<b;i++)
#define in(x, a, b) (a <= x && x < b)
#define all(c) c.begin(),c.end()
template<class T> bool chmax(T &a, const T &b) { if (a<b) { a = b; return 1; } return 0; }
template<class T> bool chmin(T &a, const T &b) { if (a>b) { a = b; return 1; } return 0; }
#define UNIQUE(v) v.erase(std::unique(v.begin(), v.end()), v.end());
const ll inf = 1000000001;
const ll INF = (ll)1e18 + 1;
const ll MOD = 1000000007;
//const ll MOD = 998244353;
const double pi = 3.14159265358979323846;
#define Sp(p) cout<<setprecision(15)<< fixed<<p<<endl;
int dx[4] = { -1,0, 1,0 }, dy[4] = { 0,1,0,-1 };
int dx2[8] = { 1,1,0,-1,-1,-1,0,1 }, dy2[8] = { 0,1,1,1,0,-1,-1,-1 };
#define fio() cin.tie(0); ios::sync_with_stdio(false);
// #define mp make_pair
//#define endl '\n'

class UnionFind {
public:
  int n;
  vi par; //親
  vi ran; //木の深さ
  vi num; //要素数
  int g; // group数
 
  UnionFind(int _n) {
    n = _n;
    g = n;
    par.resize(n); ran.resize(n); num.resize(n);
    for (int i = 0; i < n; i++) {
      par[i] = i; ran[i] = 0; num[i] = 1;
    }
  }
 
  //木の根を求める
  int find(int x) {
    if (par[x] == x) {
      return x;
    }
    else {
      return par[x] = find(par[x]);
    }
  }
 
  //xとyの属する集合を併合
  void unite(int x, int y) {
    x = find(x); y = find(y);
    int numsum = num[x] + num[y];
    if (x == y) {
      return;
    }
    if (ran[x]<ran[y]) {
      par[x] = y;
    }
    else {
      par[y] = x;
      if (ran[x] == ran[y]) {
        ran[x]++;
      }
    }
    num[x] = num[y] = numsum;
    g--;
  }
 
  //xとyが同じ集合に属するか否か
  bool same(int x, int y) {
    return find(x) == find(y);
  }
 
};


int h = 5, w = 5;
vii p2idx(h, vi(w));
vector<pii> idx2p(h*w);
vll c(h, vl(w));

ll solve() {
  UnionFind uf(25);
  ll ans = INF;

  rep (l, 16) {
    for (int r = l; r < 16; r++) {
      if (l == r and l != 0) continue;
      vll color(h, vl(w));
      for (int i = l; i < r; i++) {
        color[idx2p[i].first][idx2p[i].second] = 1;
      }
      
      rep (i, 1 << 9) {
        uf = UnionFind(25);
        rep (j, 9) {
          int u = 16 + j;
          int y = idx2p[u].first, x = idx2p[u].second;
          if (i & (1 << j)) color[y][x] = 1;
          else color[y][x] = 0;
        }
        
        rep (y, h) {
          rep (x, w) {
            if (y != h - 1) {
              if (color[y][x] == color[y + 1][x]) uf.unite(p2idx[y][x], p2idx[y + 1][x]);
            }
            if (x != w - 1) {
              if (color[y][x] == color[y][x + 1]) uf.unite(p2idx[y][x], p2idx[y][x + 1]);
            }
          }
        }
        if (uf.g != 2) continue;
        ll sa = 0;
        rep (y, h) rep (x, w) {
          if (color[y][x]) sa += c[y][x];
          else sa -= c[y][x];
        }
        chmin(ans, abs(sa));
      }
    }
  }

  return ans;
}

int main() {
  rep (i, h) rep (j, w) cin >> c[i][j];
  
  int cnt = 0;
  rep (j, 5) {
    p2idx[0][j] = cnt;
    idx2p[cnt] = pii(0, j);
    cnt++;
  }
  rep1 (i, 3) {
    p2idx[i][4] = cnt;
    idx2p[cnt] = pii(i, 4);
    cnt++;
  }
  rrep (j, 5) {
    p2idx[4][j] = cnt;
    idx2p[cnt] = pii(4, j);
    cnt++;
  }
  rrep1 (i, 3) {
    p2idx[i][0] = cnt;
    idx2p[cnt] = pii(i, 0);
    cnt++;
  }
  rep1 (i, 3) {
    rep1 (j, 3) {
      p2idx[i][j] = cnt;
    idx2p[cnt] = pii(i, j);
    cnt++;
    }
  }
  //DEBUG_MAT(p2idx);
  //DEBUG_VEC(idx2p);

  ll ans = solve();
  cout << ans << endl;
}
0