NVDIA（英伟达)一道笔试笔试编程题

NVDIA（英伟达)一道笔试笔试编程题

Given an integer number N(n = m^2 - 1),
 print sequence 0, 1, 2, 3, ..., n，in below form. 
You can only use printf(C) or cout(C++). 
For example, given n = 24(5^2 - 1), the program should output

  0    1     2    3   4

15  16  17  18   5

14  23  24  19   6

13  22  21  20   7

12  11  10   9    8

1.  What memory & computation complexity of your algorithm?  
      //你的算法的时间和空间的复杂度？

2.  Is there an algorithm to get O(1) of memory and O(n) of computation complexity

  //有没有空间复杂度为O(1),而时间复杂度为O(n)的算法？
分析：
  0   1     2    3  4
15   0    1    2   5
14   7    0    3   6
13   6    5    4   7
12 11 10    9   8

 1. 空间复杂度和时间复杂度都是O(n)的算法的实现如下：

 

#include <stdio.h>
#include <stdlib.h>
#include <math.h>

#define MAX 100000

void screw(int*a, int n);
void printArray(int*a, int n);

int main(int argc, char* argv[], char* env[])
{
int a[MAX] = {0,};
int i;
int n;

for(i=1; i<10; i++) {
n = i*i -1;
screw(a, n);
printArray(a, n);
}

return 0;
}

void screw(int* a, int n) {
int m;
int p;
int q;
int i;
int j;
int k;
int base;

m = (int)sqrt(n+1);
q = 0;
k = 0;

for(p=m; p>0; p-=2) {
base = m*q + q;

//上边
j=0;
for(i=0; i<p; i++) {
a[base+j*m+i] = k++;
}

//右边
i=p-1;
for(j=1; j<p; j++) {
a[base+j*m+i] = k++;
}

//下边
j=p-1;
for(i=p-2; i>=0; i--){
a[base+j*m+i] = k++;
}

//左边
i=0;
for(j=p-2; j>0; j--) {
a[base+j*m+i] = k++;
}

q++;
}
}

void printArray(int * a, int n) {
int i;
int m;

m = (int)sqrt(n+1);

for(i=0; i<=n; i++) {
if(i%m == 0) printf("\n");
printf("%2d ", a[i]);
}
printf("\n");

}

运行结果：



2.是否有空间复杂度为O(1), 计算法复杂度为O(n)的算法？

 //怎么求逆函数，思考中ing。。