vector二维数组与蛇形矩阵

最近在学习STL，感觉真的是好用，以前二维数组动态分配要自己写循环，还要注意释放内存。现在一个vector全解决。

int n; cin >> n; vector<int> line(n); vector<vector <int>> matrix(n, line); int m, i, j, num, flag = 1; m = n*n + 1; i = 0; j = 0; for (num = 1; num <= m / 2; num++) //生成矩阵 { matrix[i][j] = num; matrix[n - i - 1][n - j - 1] = m - num; i = i - flag; j = j + flag; if (i < 0){ i = 0; flag = -flag; } if (j < 0){ j = 0; flag = -flag; } } //输出矩阵 for (auto l : matrix) { for (auto k : l) { cout << k<<"\t"; } cout << endl; } system("pause"); return 0;