Submission #34129

---

Source Code Expand

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include <map>
#include <sstream>

using namespace std;

void erase(vector<vector<int> > &a, int i, int j) {
    int di[8] = {-1,0,1,-1,1,-1,0,1}, dj[8]={-1,-1,-1,0,0,1,1,1};
    if (i<0 || a.size() <= i || j<0 || a[0].size() <= j)return;
    if (a[i][j] == 0)return;
    a[i][j] = 0;
    for (int c=0; c<8; c++) {
        erase(a, i+di[c], j+dj[c]);
    }
}

int main() {
    int h,w;
    cin >> h >> w;
    vector<string> mat(h);
    vector<vector<int> > sumi(h, vector<int>(w)),sumj(h, vector<int>(w));
    for (int i=0; i<h; i++) {
        cin >> mat[i];
    }

    for (int i=1; i<h; i++) {
        for (int j=1; j<w; j++) {
            if (mat[i][j] == 'o') {
                sumi[i][j] = sumi[i-1][j] + 1;
                sumj[i][j] = sumj[i][j-1] + 1;
            } else {
                sumi[i][j] = 0;
                sumj[i][j] = 0;
            }
        }
    }    
    int a = 0,b = 0,c = 0;
    for (int k=142; k>0; k--) {
        for (int i=0; i<h; i++) {
            for (int j=0; j<w; j++) {
                if (sumi[i][j] == k * 5 && sumj[i][j] == k * 2 || sumi[i][j] == k * 2 && sumj[i][j] == k * 5) {
                    a++;
                    erase(sumi, i, j);
                    erase(sumj, i, j);
                }
            }
        }
        for (int i=0; i<h; i++) {
            for (int j=0; j<w; j++) {
                if (sumi[i][j] == k * 4 && sumj[i][j] == k * 1 && sumi[i+1][j] == 0 || sumi[i][j] == k * 1 && sumj[i][j] == k * 4 && sumj[i][j+1] == 0){
                    b++;
                    erase(sumi, i, j);
                    erase(sumj, i, j);
                }
            }
        }
        for (int i=0; i<h; i++) {
            for (int j=0; j<w; j++) {
                if ((sumi[i][j] == k * 3 && sumj[i][j] == k * 1 || sumi[i][j] == k * 1 && sumj[i][j] == k * 3) && (sumi[i+1][j] == 0 && sumj[i][j+1] == 0)) {
                    c++;
                    erase(sumi, i, j);
                    erase(sumj, i, j);
                }
            }
        }
    }
    cout << a << " " << b << " " << c << endl;
}

Submission Info

Judge Result