题目:给定一幅有N*N矩阵表示的图像,其中每个像素的大小为4字节,编写一个方法,将图像旋转90度。不占用额外的存储空间能否做到?
思路:假设旋转是按照顺时针方向,那么右旋90度就是左移到上,上移到右,右移到下,下移到左。将矩阵分层,由外到内或者有内到外旋转。
重点复习数组和指针的有关概念,如果是二维数组,例如int a[m][n],那么a表示的是数组指针,即作为参数传入函数应该是int (*a)[4],’*'和变量名用括号括在一起,同时要指明数组的维度。另外还有一类指针数组,即元素为指针的数组,表示法为int *b[n],n表示数组元素个数,即指针的个数,具体讨论见 指针数组和数组指针的区别。
回到问题本身,写这个函数需要的形参是什么呢?如果传入二维数组,怎么传?数组名吗?那是数组指针,肯定是不对的。如果就以数组指针为参数,那么在写函数的时候必须指明数组维度。通常来说的解决办法是以指针的指针为参数,见下面代码1。指针的指针需要在堆上申请空间,而且数组赋值非常麻烦,如果我一定要用二维数组解决这个问题呢?可以传入数组的首地址作为参数,但是要注意数组元素的偏移量问题,代码见方法2。具体讨论见二维数组作为函数参数传递剖析(C语言)(6.19更新第5种)。
//方法1
#include <iostream>
using namespace std;
void rotate(int** martrix, int n)
{
for(int layer = 0; layer < n/2; ++layer)
{
int first = layer;
int last = n - 1- layer;
for(int i = first; i < last; ++i)
{
int offset = i - first;
int top = martrix[first][i];
//左移到上
martrix[first][i] = martrix[last-offset][first];
//下移到左
martrix[last-offset][first] = martrix[last][last-offset];
//右移到下
martrix[last][last-offset] = martrix[i][last];
//上移到右
martrix[i][last] = top;
}
}
}
int main() {
int **a = new int*[4];
for(int i = 0; i < 4; ++i)
{
a[i] = new int[4];
}
int num = 1;
for(int i = 0; i < 4; ++i)
{
for(int j = 0; j < 4; ++j)
a[i][j] = num++;
}
rotate(a, 4);
for(int i = 0; i < 4; ++i)
{
for(int j = 0; j < 4; ++j)
cout << a[i][j] << " ";
}
cout << endl;
return 0;
}
方法2:
//方法2
#include <iostream>
using namespace std;
void rotate(int* martrix, int n)
{
for(int layer = 0; layer < n/2; ++layer)
{
int first = layer;
int last = n - 1- layer;
for(int i = first; i < last; ++i)
{
int offset = i - first;
int top = *(martrix+first*n+i);
//左移到上
*(martrix+first*n+i) = *(martrix+(last-offset)*n+first);
//下移到左
*(martrix+(last-offset)*n+first) = *(martrix+last*n+(last-offset));
//右移到下
*(martrix+last*n+(last-offset)) = *(martrix+i*n+last);
//上移到右
*(martrix+i*n+last) = top;
}
}
}
int main() {
int a[4][4] = {1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16};
rotate(*a, 4);
for(int i = 0; i < 4; ++i)
{
for(int j = 0; j < 4; ++j)
cout << a[i][j] << " ";
}
cout << endl;
return 0;
}