Codeforces Gym 100342J Problem J. Triatrip 三元环

#include<iostream>
#include<cstring>
#include<cstdio>
#include<bitset>
using namespace std;

const int maxn = 1555;
typedef __int64 LL;

bitset<maxn> in[maxn], out[maxn], And;
char str[maxn][maxn];

int main() {
    freopen("triatrip.in", "r", stdin);
    freopen("triatrip.out", "w", stdout);
    int n;
    scanf("%d", &n);
    for (int i = 0; i < n; i++) scanf("%s", str[i]);
    LL ans = 0;
    for (int i = 0; i < n; i++){
        for (int j = 0; j < n; j++) {
            if (str[i][j] == ‘+‘) {
                in[j][i] = true;
                out[i][j] = true;
            }
        }
    }
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            if (str[i][j] == ‘+‘) {
                And = (in[i]) & (out[j]);
                ans += And.count();
            }
        }
    }
    printf("%I64d\n", ans / 3);
    return 0;
}