1. 程式人生 > >用java實現螺旋陣列

用java實現螺旋陣列

接收陣列的行數和列數,返回正序和倒序的螺旋陣列

package cn.baokx;

public class Test {
	public static void main(String[] args) {
		printArray(getSpiralArray(5,5,false));
		System.out.println("**************");
		printArray(getSpiralArray(5,5,true));
	}
	//返回螺旋陣列
	public static int [][] getSpiralArray(int m,int n,boolean reverse){
		//定義一個長度為m*n的陣列,並按順序初始化
		int [] numArray = new int[m*n];
		for(int i = 0 ; i < numArray.length ; i++){
			if(!reverse){
				numArray[i] = (i+1);
			}else{
				numArray[i] = (numArray.length-i);
			}
		}
		//初始化陣列下標
		int foot = 0;
		
		//宣告螺旋陣列
		int[][] array = new int[m][n];
		//計算"層數",以m和n中較小的數為準
		int layer = m<n?m:n;
		layer = (layer%2==1)?(layer / 2 + 1):(layer / 2);
		
		// 從外層到裡層迴圈
		for (int i = 0; i < layer; i++) { 
			//從左到右
			for (int j = i; j < n - i; j++) {
				array[i][j] = numArray[foot++];
				if(foot>=m*n){
					return array;
				}
			}
			// 從上到下
			for (int j = i + 1;  j < m - i; j++) {
				array[j][n - i - 1] = numArray[foot++];
				if(foot>=m*n){
					return array;
				}
			}
			// 從右到左
			for (int j = n - i - 2; j >= i; j--) {
				array[m - i - 1][j] = numArray[foot++];
				if(foot>=m*n){
					return array;
				}
			}
			// 從下到上
			for (int j = m - i - 2; j > i; j--) {
				array[j][i] = numArray[foot++];
				if(foot>=m*n){
					return array;
				}
			}
		}
		return array;
	}
	//列印二維陣列
	public static void printArray(int [][] array){
		for (int i = 0; i < array.length; i++) {
			for (int j = 0; j < array[0].length; j++) {
	  			if(array[i][j]<10){
	  				System.out.print("0");
	  			}
	  			System.out.print(array[i][j] + " ");
			}
			System.out.println();
		} 
	}
}