XJOI 1-21 c++ 友好数对

友好数对

题目描述：

读入一个n*m的矩阵,定义两个数是友好的，当且仅当他们位置相邻（上下左右）且值相同。求友好数对个数。

输入格式：

第一行两个整数n，m

之后n行，每行m个0到100间的整数

输出格式：

一个整数，表示答案。

样例输入1：

5 4

3 3 3 4

2 0 0 3

0 3 1 4

3 4 3 3

1 0 3 3

样例输出1：

7

约定：

1<=n，m<=500

#include <iostream>
using namespace std;
int main()
{
    int n, m;
    cin >> n >> m;
    int a[n][m];
    for (int i = 0; i < n; i++)
        for (int j = 0; j < m; j++)
        {
            cin >> a[i][j];
        }
    int cnt = 0;
    
    for (int i = 0; i < n; i++)
        for (int j = 1; j < m; j++)
        {
            if (j > 0)
            {
                if (a[i][j] == a[i][j - 1])        
                    cnt++;
            }
            if (i > 0)
            {
                if (a[i][j] == a[i - 1][j])
                    cnt++;
            }

        }
    cout << cnt << endl;
    return 0;
}

img