CCF CSP 201503-1 图像旋转 （降维）

题目链接：http://118.190.20.162/view.page?gpid=T27

问题描述

试题编号：	201503-1
试题名称：	图像旋转
时间限制：	5.0s
内存限制：	256.0MB
问题描述：	问题描述 　　旋转是图像处理的基本操作，在这个问题中，你需要将一个图像逆时针旋转90度。 　　计算机中的图像表示可以用一个矩阵来表示，为了旋转一个图像，只需要将对应的矩阵旋转即可。 输入格式 　　输入的第一行包含两个整数n, m，分别表示图像矩阵的行数和列数。 　　接下来n行每行包含m个整数，表示输入的图像。 输出格式 　　输出m行，每行包含n个整数，表示原始矩阵逆时针旋转90度后的矩阵。 样例输入 2 3 1 5 3 3 2 4 样例输出 3 4 5 2 1 3 评测用例规模与约定 　　1 ≤ n, m ≤ 1,000，矩阵中的数都是不超过1000的非负整数。

我的思路是：先将二维数组降维并存入长度为n*m的一维数组，然后按照以列逆序以行正序的方式逐个存入新的二维数组中，最后输出新的二维数组

#include <iostream>
#include <algorithm>
#include <cmath> 
#include <string>
#include <cstring>
using namespace std;
int n,m,sum;
int a[1000005];
int b[1005][1005];
int main()
{
    while(cin>>n>>m){
        sum=n+m;
        for(int i=0;i<n*m;i++) cin>>a[i];
        int t=0,flag=0;
        for(int j=0;j<n;j++){
            for(int i=m-1;i>=0;i--){
                b[i][j]=a[t++];
                if(t==n*m){
                    flag=1;
                    break;
                }
            }
            if(flag) break;
        }
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(j==0) cout<<b[i][j];
                else cout<<" "<<b[i][j];
            }
            cout<<endl;
        }
    }
    return 0;
}