結果

問題 No.861 ケーキカット
ユーザー mamekin
提出日時 2019-08-10 17:38:37
言語 C++14
(gcc 8.3.0)
結果
AC  
実行時間 560 ms
コード長 2,503 Byte
コンパイル時間 1,072 ms
使用メモリ 5,048 KB
最終ジャッジ日時 2019-10-25 14:41:28

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
sample1.txt AC 559 ms
5,044 KB
sample2.txt AC 558 ms
5,040 KB
sample3.txt AC 560 ms
5,048 KB
test1.txt AC 559 ms
5,048 KB
test2.txt AC 558 ms
5,044 KB
test3.txt AC 556 ms
5,040 KB
test4.txt AC 555 ms
5,048 KB
test5.txt AC 555 ms
5,044 KB
test6.txt AC 556 ms
5,044 KB
test7.txt AC 557 ms
5,044 KB
test8.txt AC 557 ms
5,044 KB
test9.txt AC 557 ms
5,044 KB
test10.txt AC 559 ms
5,040 KB
test11.txt AC 560 ms
5,044 KB
test12.txt AC 556 ms
5,040 KB
test13.txt AC 559 ms
5,048 KB
test14.txt AC 555 ms
5,048 KB
test15.txt AC 559 ms
5,040 KB
test16.txt AC 556 ms
5,044 KB
test17.txt AC 560 ms
5,044 KB
test18.txt AC 559 ms
5,048 KB
test19.txt AC 559 ms
5,044 KB
test20.txt AC 556 ms
5,048 KB
テストケース一括ダウンロード

ソースコード

diff #
#define _USE_MATH_DEFINES
#include <cstdio>
#include <iostream>
#include <sstream>
#include <fstream>
#include <iomanip>
#include <algorithm>
#include <cmath>
#include <complex>
#include <string>
#include <vector>
#include <array>
#include <list>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <bitset>
#include <numeric>
#include <limits>
#include <climits>
#include <cfloat>
#include <functional>
#include <iterator>
#include <memory>
#include <regex>
using namespace std;

const int N = 5;
const long long INF = LLONG_MAX / 2;
const int dy[] = {0, 1, 0, -1};
const int dx[] = {-1, 0, 1, 0};

bool check(const vector<vector<int> >& cake)
{
    for(int i=0; i<2; ++i){
        int sy = -1;
        int sx = -1;
        for(int y=0; y<N; ++y){
            for(int x=0; x<N; ++x){
                if(cake[y][x] == i){
                    sy = y;
                    sx = x;
                }
            }
        }
        if(sy == -1)
            continue;

        vector<vector<bool> > used(N, vector<bool>(N, false));
        queue<pair<int, int> > q;
        used[sy][sx] = true;
        q.push(make_pair(sy, sx));
        while(!q.empty()){
            int y, x;
            tie(y, x) = q.front();
            q.pop();
            for(int j=0; j<4; ++j){
                int y2 = y + dy[j];
                int x2 = x + dx[j];
                if(!(0 <= y2 && y2 < N && 0 <= x2 && x2 < N))
                    continue;
                if(!used[y2][x2] && (cake[y2][x2] == i || cake[y2][x2] == -1)){
                    used[y2][x2] = true;
                    q.push(make_pair(y2, x2));
                }
            }
        }

        for(int y=0; y<N; ++y){
            for(int x=0; x<N; ++x){
                if(cake[y][x] == i && !used[y][x])
                    return false;
            }
        }
    }

    return true;
}

long long solve(const vector<int>& val, vector<vector<int> >& cake, int k, long long diff)
{
    if(!check(cake))
        return INF;
    if(k == N * N)
        return abs(diff);

    long long ans = INF;
    for(int i=0; i<2; ++i){
        long long diff2 = diff + val[k] * (i == 0 ? 1 : -1);
        cake[k/5][k%5] = i;
        ans = min(ans, solve(val, cake, k+1, diff2));
        cake[k/5][k%5] = -1;
    }
    return ans;
}

int main()
{
    vector<int> val(N*N);
    for(int i=0; i<N*N; ++i)
        cin >> val[i];

    vector<vector<int> > cake(N, vector<int>(N, -1));
    cout << solve(val, cake, 0, 0) << endl;

    return 0;
}
0