← Back to List

1420번: 학교 가지마! ↗

Solutions

C++14
3.7 KB | 3793 chars
#include <bits/stdc++.h>

#define for1(s,n) for(int i = s; i < n; i++)
#define for1j(s,n) for(int j = s; j < n; j++)
#define foreach(k) for(auto i : k)
#define foreachj(k) for(auto j : k)
#define pb(a) push_back(a)
#define sz(a) a.size()

using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef vector <int> iv1;
typedef vector <vector<int>> iv2;
typedef vector <ll> llv1;
typedef unsigned int uint;
typedef vector <ull> ullv1;
typedef vector <vector <ull>> ullv2;


#define MAX_V 20011
#define INF (ll)1e18

struct Edge {
    ll v, capacity, rev;
    Edge(ll v, ll capacity, ll rev): v(v), capacity(capacity), rev(rev) {}
};

vector<Edge> vt[MAX_V];
ll level[MAX_V];
ll work[MAX_V];
ll SINK;
ll SRC;


void addEdge(ll start, ll end, ll capacity) {
    vt[start].emplace_back(end, capacity, (ll)vt[end].size());
    vt[end].emplace_back(start, 0, (ll)vt[start].size()-1); 
}

// 레벨 그래프 만드는 BFS
bool bfs() {
    memset(level, -1, sizeof(level));        //레벨 그래프 초기화
    queue <ll> q;
    level[SRC] = 0;
    q.push(SRC);

    while(!q.empty()){
        int here = q.front(); q.pop();
        for (auto i : vt[here]) {
            ll there = i.v;
            if(level[there] == -1 && i.capacity > 0) {
                level[there] = level[here] + 1;
                q.push(there);
            }
        }
    }
    return level[SINK] != -1;
}


ll dfs(ll here, ll crt_capacity) {
    if(here == SINK) return crt_capacity;

    for(ll &i = work[here]; i < vt[here].size(); i++) {
        ll there = vt[here][i].v;
        ll capacity = vt[here][i].capacity;

        if(level[here] + 1 == level[there] && capacity > 0) {
            ll next_capacity = dfs(there, min(crt_capacity, capacity));

            if(next_capacity > 0) {
                vt[here][i].capacity -= next_capacity;
                vt[there][vt[here][i].rev].capacity += next_capacity;
                return next_capacity;
            }
        }
    }
    return 0;
}

ll dinic() {
    ll ret = 0;
    while(bfs()) {
        memset(work, 0, sizeof(work));

        while(1) {
            ll flow = dfs(SRC, INF);
            if(!flow) break;
            ret += flow;
        }
    }
    return ret;
}

ll N, M;
string m[110];

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    
    cin >> N >> M;
    for1(0, N) {
        cin >> m[i];
    }
    for1(0, N) {
        for1j(0, M) {
            if(m[i][j] == 'K') SINK = i *M + j;
            if(m[i][j] == 'H') SRC = i *M + j;
        }
    }

    if(N == 1) {
        if(M==1 || M == 2) {
            cout << "-1";
            return 0;
        }
    }
    if(M == 1 && N == 2) {
        cout << "-1";
        return 0;
    }

    if(((SINK/M == SRC/M)  && abs((SINK%M)-(SRC%M)) == 1) || ((SINK%M == SRC%M)  && abs((SINK/M)-(SRC/M)) == 1)) {
        cout << "-1";
        return 0;
    }

    for1(0, N) {
        for1j(0, M) {
            if(i*M+j == SRC || i*M +j == SINK) {
                addEdge(i*M+j, i*M+j + N*M, INF);
            }
            else {
                addEdge(i*M+j, i*M+j + N*M, 1);
            }
            if(i+1<N) {
                if(m[i][j]!='#' && m[i+1][j]!='#') {
                    addEdge(i*M+j+N*M, (i+1)*M+j,1);
                }
            }
            if(j +1 < M) {
                if(m[i][j]!='#' && m[i][j+1]!='#') {
                    addEdge(i*M+j+N*M, i*M+j+1,1);
                } 
            }
            if(i-1 >= 0) {
                if(m[i][j]!='#' && m[i-1][j]!='#') {
                    addEdge(i*M+j+N*M, (i-1)*M+j,1);
                }
            }
            if(j -1 >= 0) {
                if(m[i][j]!='#' && m[i][j-1]!='#') {
                    addEdge(i*M+j+N*M, i*M+j-1,1);
                }
            }
        }
    }
    cout << dinic();
    
}