Leetcode566. Reshape the Matrix

Posted 2020-09-15 vector11248

	public int[][] func(int[][] nums,int r,int c){
		int m=nums.length;
		int k=0;
		int n=nums[0].length;
		  if(r*c!=m*n){
			return nums;
		}
		  int[][] res = new int[r][c];
		  for(int i=0;i<r;i++){
			  for(int j=0;j<c;j++,k++){
				  res[i][j]=nums[k/n][k%n];
			  }
		  }
		  return res;
		  
	}