循环通过二维数组(对角线) c++

#include <iostream>

int main() {
    int arr[3][3] = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};

    int rows = 3;
    int cols = 3;

    // 遍历对角线上方的元素
    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < cols - i; j++) {
            std::cout << arr[i][j] << " ";
        }
    }

    // 遍历对角线下方的元素
    for (int i = 1; i < rows; i++) {
        for (int j = i; j < cols; j++) {
            std::cout << arr[j][i] << " ";
        }
    }

    return 0;
}