Diagonal

#include <iostream>
#include <iomanip>
using namespace std;

int element(int i, int j, int n, int m) {
    int p = 0;
    int j1 = j;
    if (i == 0) {
        for (i; i < n; i++) {
            for (int k = j; k > 0; k--) {
                p++;
            }
            j--;
            if (j == 0) {
                break;
            }
        }
    }
    else if (i >= 1) {
        for (int ii = i; ii >= 0; ii--) {
            if (ii == i) {
                for (int k = j; k > 0; --k) {
                    p++;
                }
            }
            else {
                for (int k = j; k >= 0; --k) {
                    p++;
                }
            }
            if (j + 1 <= m - 1) {
                j++;
            }
        }
        if (i + 1 <= n) {
        j1 = j1 - 1;
            for (int ii = i + 1; ii < n; ii++) {
                for (int k = j1; k > 0; --k) {
                    p++;
                }
                if (j1 - 1 >= 0) {
                    j1--;
                }
            }
        }
    }
    return (p + 1);
}

int task13_2()
{
    int n, m;
    cin >> n >> m;
    int count = 0;
    int p = 0;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            cout << setw(4) << element(i , j, n, m);
        }
        cout << endl;
    }
    return 0;
}

int main()
{
    task13_2();
    system("PAUSE");
    return 0;
}