极简题解|复杂的线性代数问题

总题解
难点在于遍历主斜线上方一条线有很多细节，注意区分；
遇到非0直接输出并return 0

#include<iostream>
using namespace std;
int a,b;
int main(){
    cin>>a;
    for(int i=0;i<a;i++){
        for(int j=0;j<a;j++){
            cin>>b;
            if(i+1==j){
                if(b!=0){
                    cout<<"NO";
                    return 0;
                }
            }
        }
    }
    cout<<"YES";
}