这是一道leecode美团的一道周赛题,属于中等:----------->原题
给你一个大小为 m x n 的二维整数网格 grid 和一个整数 x 。每一次操作,你可以对 grid 中的任一元素 加 x 或 减 x 。单值网格 是全部元素都相等的网格。返回使网格化为单值网格所需的 最小 操作数。如果不能,返回 -1 。
输入:grid = [[2,4],[6,8]], x = 2
输出:4
解释:可以执行下述操作使所有元素都等于 4 :
- 2 加 x 一次。
- 6 减 x 一次。
- 8 减 x 两次。
共计 4 次操作。
当我做这道题的时候,觉得就应该先把二维数组转化为一维数组
int m=grid.length;
int n=grid[0].length;
int[] Q=new int[m*n];
int t=0;
//二维数组转化为一维数组
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
Q[t++]=grid[i][j];
}
}
要得到最小操作数,对于一个一维数组而言,那就应该是中间的数到两边的情况下才会使得操作数最小
//排序
Arrays.sort(Q);
//求中位数
int md=Q[m*n/2];
要想让某个数通过+X或-X变成中位数,那这个数与中位数的差值就应该是X的倍数。所以:
int sum=0;
for(int i=0;i<Q.length;i++){
int d=Math.abs(Q[i]-md);
if(d%x!=0){ //是否是x的倍数
return -1;
}
sum=sum+d/x;
}
附上完整代码:
class Solution {
public int minOperations(int[][] grid, int x) {
int m=grid.length;
int n=grid[0].length;
int[] Q=new int[m*n];
int t=0;
//二维数组转化为一维数组
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
Q[t++]=grid[i][j];
}
}
//排序
Arrays.sort(Q);
//求中位数
int md=Q[m*n/2];
int sum=0;
for(int i=0;i<Q.length;i++){
int d=Math.abs(Q[i]-md);
if(d%x!=0){
return -1;
}
sum=sum+d/x;
}
return sum;
}
}