找数字 算法题

题目描述

给一个二维数组nums，对于每一个元素num[i]，找出距离最近的且值相等的元素，输出横纵坐标差值的绝对值之和，如果没有等值元素，则输出-1。

例如：

输入数组nums为

0 3 5 4 2
2 5 7 8 3
2 5 4 2 4

对于 num[0][0] = 0，不存在相等的值。

对于 num[0][1] = 3，存在一个相等的值，最近的坐标为num[1][4]，最小距离为4。

对于 num[0][2] = 5，存在两个相等的值，最近的坐标为num[1][1]，故最小距离为2。

…

对于 num[1][1] = 5，存在两个相等的值，最近的坐标为num[2][1]，故最小距离为1。

…

故输出为

-1 4  2  3 3
 1 1 -1 -1 4
 1 1  2  3 2

输入描述

输入第一行为二维数组的行

输入第二行为二维数组的列

输入的数字以空格隔开。

输出描述

数组形式返回所有坐标值。

备注

示例一

输入

3
5
0 3 5 4 2
2 5 7 8 3
2 5 4 2 4

输出

[[-1, 4, 2, 3, 3], [1, 1, -1, -1, 4], [1, 1, 2, 3, 2]]

代码演示

public static void main(String[] args) {
//控制台输入
        Scanner sc = new Scanner(System.in);
        //行数
        int m = sc.nextInt();
        //列数
        int n = sc.nextInt();
        //输入数组的数值
        int[][] nums = new int[m][n];
        for(int i = 0; i < m ; i++){
            for(int j = 0; j<n;j++){
                nums[i][j]=sc.nextInt();
            }
        }
      int[][] solution = solution(m, n, nums);
       for(int i =0;i<m;i++){
           for(int j = 0 ;j<n;j++){
               System.out.print(solution[i][j]+" ");
           }
           System.out.println();
       }

    }

    public static int[][] solution(int m,int n,int[][] nums){
    //用于记录相同数字的最短距离
        int[][] dis = new int[m][n];
        //遍历所有的数字
        for(int i = 0;i<m;i++){
           for(int j = 0;j<n;j++){
           //提取每个数字进行比较并计算最短距离
              int num = nums[i][j];
              int min = m*n;
              for(int x = 0;x<m;x++){
                  for(int y = 0; y<n;y++){
                      if(num==nums[x][y] && ( x!=i || y!=j )){
                          int real = Math.abs(x-i)+Math.abs(y-j);
                          min = Math.min(min,real);
                      }
                  }
              }
              if(min!=m*n){
                  dis[i][j] = min;
              }else{
              //没有最短距离的赋值为-1
                  dis[i][j] = -1;
              }

           }
        }
        return dis;
    }

测试