-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathtriangles.cpp
90 lines (83 loc) · 1.67 KB
/
triangles.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include <bits/stdc++.h>
#define endl '\n'
#define eat cin
#define moo cout
#define int long long
using namespace std;
int N, ans = 0, D[600][300];
string S[300];
void flip(){
for(int i = 0; i < N; i++){
reverse(S[i].begin(), S[i].end());
}
}
void rotate(){
flip();
for(int i = 0; i < N; i++){
for(int j = 0; j < i; j++){
swap(S[i][j], S[j][i]);
}
}
}
void solve(){
for(int i = 0; i < 2*N; i++){
for(int j = 0; j < N; j++) D[i][j] = 0;
}
for(int i = N-1; i >= 0; i--){
for(int j = 0; j < N; j++){
D[i+j][j] = (S[i][j] == '*');
if(i < N-1 && j > 0) D[i+j][j] += D[i+j][j-1];
}
}
for(int i = 0; i < 2*N; i++){
for(int j = 1; j < N; j++){
D[i][j] = max(D[i][j], D[i][j-1]);
}
}
for(int i = 0; i < N; i++){
for(int j = 0; j < N; j++){
if(S[i][j] != '*') continue;
int x = i, y = j;
for(int k = 2; 1; k+=2){
x--, y++;
if(x < 0 || y >= N || i+j-k < 0) break;
if(S[x][y] != '*') continue;
int cur = D[i+j-k][j];
int endy = j - k/2;
if(endy > 0) cur -= D[i+j-k][endy-1];
ans += cur;
}
}
}
}
void elim(){
for(int i = 0; i < N; i++){
for(int j = 0; j < N; j++){
if(S[i][j] != '*') continue;
for(int k = j+2; k < N; k += 2){
if(S[i][k] != '*') continue;
int med = (j+k)/2;
int d = (k-j)/2;
int x1 = i-d, x2 = i+d;
if(x1 >= 0 && S[x1][med] == '*') ans--;
if(x2 < N && S[x2][med] == '*') ans--;
}
}
}
}
int32_t main(){
eat.tie(0) -> sync_with_stdio(0);
freopen("triangles.in", "r", stdin); freopen("triangles.out", "w", stdout);
eat >> N;
for(int i = 0; i < N; i++){
eat >> S[i];
}
for(int j = 0; j < 4; j++){
solve();
rotate();
}
elim();
rotate();
elim();
moo << ans << endl;
}