oj比赛「25年2月- A组（萌新）」D灯光设计「答案」

#include <bits/stdc++.h>
using namespace std;
int n,c,i;
char a[310][310]; 
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cin>>a[i][j];
        }
    }
    int cnt,ans=0;
    for(i=1;i<=n;i++){
        for(int j=1+i;j<=n;j++){
            cnt=0;
            for(int k=1;k<=n;k++){
                    if(a[i][k]=='1'&&a[j][k]=='1'){
                        cnt++;
                    }
            }
            if(cnt>=2){
                ans+=cnt*(cnt-1)/2;
            }
        }
    }
    
    
    cout<<ans;
    return 0;
}