Code Kata:螺旋矩陣 javascript實現

如圖所示，就是一個5*5的螺旋矩陣

我的思路如下：

第一步：拆分“層”數組

把矩陣根據層數分成N個連續的自然數組，根據如果每一層寬度為n的話，那麼每一層一共就有4(n-1)個數字，且當n=1時個數為1

拆分數字代碼

 1 function splitNumbers(n,m){ /*將總數按照每一層拆分為N個數組*/
 2 
 3     var arr = [];
 4 
 5     if(n == 1){
 6 
 7         arr[0] = 1 + m;
 8 
 9     }
10     else{
11 
12         for(var i=0;i<4*n-4;i++){
13 
14             arr[i] = i + 1 + m;
15 
16         }
17     }
18 
19     return arr;
20 }

迴圈調用，n每次減2

 1  var m = 0;
 2 
 3     while(n > 0){
 4 
 5         layerArray[count] = splitNumbers(n,m);
 6 
 7         n = n - 2;
 8 
 9         m = layerArray[count][layerArray[count].length - 1];
10 
11         count++;
12     }

第二部：將其組裝為一個“沙漏”數組

將第一步拆分好的數組取第1到n，以及2n-1到3n-2，分別組裝到“沙漏”數組的兩端，其中“沙漏”底部的數組需要倒序排列

部分代碼

 1  for(var i=0;i<totalLayers;i++){/*組裝成漏斗數組*/
 2 
 3         var index = i >= layerArray.length ? totalLayers - i - 1 : i;
 4 
 5         var cloneArray = layerArray[index].concat();
 6 
 7         if(i < totalLayers / 2){
 8 
 9             spiralArray[i] = cloneArray.splice(0,totalLayers - index * 2);
10         }
11         else{
12 
13             spiralArray[i] = cloneArray.splice(2 * (totalLayers - 2 * i) - 2,totalLayers - index * 2);
14 
15             spiralArray[i] = spiralArray[i].reverse();
16         }
17     }

第三部：把“沙漏”數組填充為完整的矩陣

“沙漏”數組，第1層和倒數第n層不需要填充，從第2層開始，依次向兩端填充每個“層”數組中的第n+m位以及4(n-1)-m，m位“沙漏”數組的層數

部分代碼

 1     for(var i=0;i<spiralArray.length;i++){/*填充漏斗*/
 2 
 3         var index = i >= layerArray.length ? totalLayers - i - 1 : i;
 4 
 5         for(var j=0;j<index;j++){
 6 
 7             var cloneArray = layerArray[j].concat();
 8 
 9             spiralArray[i].splice(j,0,cloneArray[cloneArray.length - i + j]);
10 
11             spiralArray[i].splice(spiralArray[i].length - j,0,cloneArray[totalLayers - 2 * j + i - j - 1]);
12 
13         }
14     }

 全部代碼 github地址:https://github.com/yux357/my-code-kata/blob/master/spiralNumbers.js

 1 function splitNumbers(n,m){ /*將總數按照每一層拆分為N個數組*/
 2 
 3     var arr = [];
 4 
 5     if(n == 1){
 6 
 7         arr[0] = 1 + m;
 8 
 9     }
10     else{
11 
12         for(var i=0;i<4*n-4;i++){
13 
14             arr[i] = i + 1 + m;
15 
16         }
17     }
18 
19     return arr;
20 }
21 
22 function spiralNumbers(n){
23 
24     var layerArray = [];
25 
26     var spiralArray = [];
27 
28     var totalLayers = n;
29 
30     var count = 0;
31 
32     var m = 0;
33 
34     while(n > 0){
35 
36         layerArray[count] = splitNumbers(n,m);
37 
38         n = n - 2;
39 
40         m = layerArray[count][layerArray[count].length - 1];
41 
42         count++;
43     }
44 
45     for(var i=0;i<totalLayers;i++){/*組裝成漏斗數組*/
46 
47         var index = i >= layerArray.length ? totalLayers - i - 1 : i;
48 
49         var cloneArray = layerArray[index].concat();
50 
51         if(i < totalLayers / 2){
52 
53             spiralArray[i] = cloneArray.splice(0,totalLayers - index * 2);
54         }
55         else{
56 
57             spiralArray[i] = cloneArray.splice(2 * (totalLayers - 2 * i) - 2,totalLayers - index * 2);
58 
59             spiralArray[i] = spiralArray[i].reverse();
60         }
61     }
62 
63     for(var i=0;i<spiralArray.length;i++){/*填充漏斗*/
64 
65         var index = i >= layerArray.length ? totalLayers - i - 1 : i;
66 
67         for(var j=0;j<index;j++){
68 
69             var cloneArray = layerArray[j].concat();
70 
71             spiralArray[i].splice(j,0,cloneArray[cloneArray.length - i + j]);
72 
73             spiralArray[i].splice(spiralArray[i].length - j,0,cloneArray[totalLayers - 2 * j + i - j - 1]);
74 
75         }
76     }
77 
78     return spiralArray;
79 }