CFF第二题D34

CCF-CSP第34次认证第二题——矩阵重塑（其二）【需反复思考学习！！！】 - 技术栈

D34

这道题主要分类讨论的思想，顺便复习了一下（一）的思想

代码：

//矩阵重塑 int n,m,t; cin>>n>>m>>t; vector<vector<int>> ans(n, vector<int>(m)); //n*m; for(int i=0;i<n;i++){ for(int j=0;j<m;j++){ cin>>ans[i][j]; } } while(t--){ int op,a,b; cin>>op>>a>>b; if(op==2){//转置 //int tmp; vector<vector<int>>ret(m,vector<int>(n));//格式 for(int i=0;i<n;i++){ for(int j=0;j<m;j++){ ret[j][i]=ans[i][j]; //tmp=ans[i][j]; //ans[i][j]=ans[j][i]; //ans[j][i]=tmp; } } ans=ret; swap(n, m); } if(op==3){ cout<<ans[a][b]<<endl; } if(op==1){ vector<vector<int>> ret(a,vector<int>(b));//格式 for(int i=0;i<a;i++){ for(int j=0;j<b;j++){ int K=i*b+j; int ori_i=K/m; int ori_j=K%m; //K=i*m+j+1;K=I*M+J; ret[i][j]=ans[ori_i][ori_j]; } } // 修正3：更新原矩阵和维度，后续操作基于新矩阵 ans = ret; n = a; m = b; } }

运行结果：

复盘：

1.自己成功推导出来了一的那个公式，有进步

2.新学会了矩阵交换的写法：我还以为要手写两行for呢

ans = ret; n = a; m = b;

3.在矩阵置换那里犯了一个愚蠢的错误

if(op==2){//转置 int tmp; for(int i=0;i<n;i++){ for(int j=0;j<m;j++){ tmp=ans[i][j]; ans[i][j]=ans[j][i]; ans[j][i]=tmp; } } swap(n, m); }

一开始想用中间变量直接转置，还在洋洋得意，发现纯是大傻春了。两边转置又都转回去了

后面想着那就转上三角吧，结果发现又没说是方阵，没招了。

附：上三角矩阵转置，直接原地转就可以

if (op == 2) { // 方阵原地转置（核心逻辑） int tmp; // 仅用1个临时变量 // 遍历上三角区域：i < j，避免重复交换 for (int i = 0; i < n; ++i) { for (int j = i + 1; j < n; ++j) { // 三步交换：仅用临时变量tmp tmp = ans[i][j]; ans[i][j] = ans[j][i]; ans[j][i] = tmp; } } cout << "方阵原地转置完成！" << endl; // 方阵转置后n=m不变，无需swap(n,m) }

以上完毕