【PTA】7-3 判断上三角矩阵 (15分)

#include 
int f(int n){
	int sum=0;
	while(n>0){
		sum+=n;
		n--;
	} 
	return sum;
}
int main()
{
	int T;
    scanf("%d",&T);
    int n;
    for(int i=0;i<T;i++){
    	scanf("%d",&n);
    	int arr[n][n],count=0;
    	for(int j=0;j<n;j++){
    		for(int k=0;k<n;k++){
    			scanf("%d",&arr[j][k]);
    			if(j>k&&arr[j][k]==0) count++;
			}
		}
		if(count==f(n-1)) printf("YES");
		else printf("NO");
		if(i!=T-1) printf("\n");
	}
    return 0;
}