Submission #3981584

---

Source Code Expand

#include <string>
#include <vector>
#include <iostream>
using namespace std;
long long solve(vector<string> S) {
	int H = S.size(), W = S[0].size();
	long long ans = 0;
	for (int i = 0; i <= H + W - 2; ++i) {
		for (int j = i + 2; j <= H + W - 2; j += 2) {
			int B = (j - i) / 2;
			for (int k = 0; k < H - B; ++k) {
				if (i - k >= W || i - k - B < 0 || S[k][i - k] == '.' || S[k + B][i - k - B] == '.') continue;
				for (int l = k + B; l <= k + 2 * B && l < H; ++l) {
					ans += (0 <= j - l && j - l < W && S[l][j - l] == '#' ? 1 : 0) * (l == k + B || l == k + 2 * B ? 1 : 2);
				}
			}
		}
	}
	return ans;
}
vector<string> rotate(vector<string> S) {
	int H = S.size(), W = S[0].size();
	vector<string> ans(W, string(H, '?'));
	for (int i = 0; i < H; ++i) {
		for (int j = 0; j < W; ++j) {
			ans[j][H - i - 1] = S[i][j];
		}
	}
	return ans;
}
int main() {
	int H, W;
	cin >> H >> W;
	vector<string> S(H);
	for (int i = 0; i < H; ++i) {
		cin >> S[i];
	}
	long long ans = 0;
	for (int i = 0; i < 4; ++i) {
		ans += solve(S);
		S = rotate(S);
	}
	cout << ans / 2 << endl;
	return 0;
}

Submission Info

Judge Result