PAT 乙级 1097

依旧花时间的一集。

创建二维数组并初始化为 0 的写法：vector<vector<int>> a(n + 1, vector<int>(n + 1));

#include<bits/stdc++.h> using namespace std; int main() { int n,k,x; cin >> n >> k >> x; vector<vector<int>> a(n + 1, vector<int>(n + 1)); for(int i = 1; i <= n; i ++) for(int j = 1; j <= n; j ++) cin >> a[i][j]; for(int i = 1; i <= n; i += 2) { int yi = 0; if( (i+1) / 2 % k == 0) yi = k; else yi = (i+1) / 2 % k; for(int j = n; j > yi; j --) a[i][j] = a[i][j - yi]; for(int j = 1; j <= yi; j ++) a[i][j] = x; } int flag = 0; for(int j = 1; j <= n; j ++) { int sum = 0; for(int i = 1; i <= n; i ++) sum += a[i][j]; if(flag == 1) cout << " "; flag = 1; cout << sum; } return 0; }