【码蹄集新手村 600 题】判断是否为单位矩阵

最新推荐文章于 2024-10-01 05:04:32 发布

Sandm *

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量312

点赞数

分类专栏：码蹄集新手村 600 题目

本文链接：https://blog.csdn.net/m0_64075307/article/details/126396609

版权

leetcode 算法职场和发展

码蹄集新手村 600 题目专栏收录该内容

76 篇文章 18 订阅

订阅专栏

题目链接：

解题思路：

单位矩阵：

一个矩阵中的对角线上的值均为 1，

非对角线上的值均为 0.

注意：

求和变量 sum 的值

参考代码：

#include<bits/stdc++.h> 

using namespace std;

int main( )
{
    int arr[3][3] = {0};
    int flag = 3;

    //int sum = 0;sum不能放在该位置上

    for(int i = 0;i < 3;i++)
    {
        int sum = 0;

        for(int j = 0;j < 3;j++)
        {
            cin >> arr[i][j];

            sum += arr[i][j];
        }

        if((arr[i][i] == 1) && (sum == 1))
        {
            flag--;
        }
       
    }

    if (flag == 0)
    {
        cout << "YES" << endl;
    }
    else
    {
        cout << "NO" << endl;
    }
    
    return 0;
}