AI面板识别 - 华为OD统一考试（java）

题目描述

AI识别到面板上有N（1 ≤ N ≤ 100）个指示灯，灯大小一样，任意两个之间无重叠。

由于AI识别误差，每次别到的指示灯位置可能有差异，以4个坐标值描述AI识别的指示灯的大小和位置(左上角x1,y1，右下角x2,y2)，

请输出先行后列排序的指示灯的编号，排序规则：

1. 每次在尚未排序的灯中挑选最高的灯作为的基准灯；
2. 找出和基准灯属于同一行所有的灯进行排序。两个灯高低偏差不超过灯半径算同一行（即两个灯坐标的差 ≤ 灯高度的一半）。

输入描述

第一行为N，表示灯的个数 接下来N行，每行为1个灯的坐标信息，格式为：

编号 x1 y1 x2 y2

• 编号全局唯一
• 1 ≤ 编号 ≤ 100
• 0 ≤ x1 < x2 ≤ 1000
• 0 ≤ y1 < y2 ≤ 1000

输出描述

排序后的编号列表，编号之间以空格分隔

示例1

输入：
5
1 0 0 2 2
2 6 1 8 3
3 3 2 5 4
5 5 4 7 6
4 0 4 2 6

输出：
1 2 3 4 5

题解

题目就是一道模拟排序题。

其实就是先按行，再按列进行排序，这里的行不是严格的行（在一定范围内可以认为是同行）。数据量不大可以直接按照题目要求进行模拟得到答案。使用map分别对编号和坐标进行存储，再根据坐标分别排序即可。

java代码

package odTest;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Scanner;

public class AIPanelIdentifi {
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int num = Integer.parseInt(scanner.nextLine());
		
		Map<Integer,int[]> coorMap = new HashMap<>();
		for(int i=0;i<num;i++) {
			int[] coors = Arrays.stream(scanner.nextLine().split(" ")).mapToInt(Integer::parseInt).toArray();
			int[] coor = new int[2];
			coor[0] = (coors[3]-coors[1])/2+coors[1];
			coor[1] = (coors[4]-coors[2])/2+coors[2];
			coorMap.put(coors[0], coor);
		}
//		for(Map.Entry<Integer, int[]> entry: coorMap.entrySet()) {
//			System.out.println(entry.getKey()+":"+Arrays.toString(entry.getValue()));
//		}
		List<Map.Entry<Integer, int[]>> list = new ArrayList<Map.Entry<Integer, int[]>>(coorMap.entrySet()); //转换为list

		Collections.sort(list,new Comparator<Map.Entry<Integer, int[]>>() {

			@Override
			public int compare(Entry<Integer, int[]> o1, Entry<Integer, int[]> o2) {
				// TODO Auto-generated method stub
				int result = 0;
				int xMinus = o1.getValue()[0]-o2.getValue()[0];
				int yMinus = o1.getValue()[1]-o2.getValue()[1];
				if(o1.getValue()[1]<o2.getValue()[1]) {
					return yMinus;
				}
				if(o1.getValue()[1] == o2.getValue()[1]) {
					return xMinus;
				}
				return yMinus;
			}
		});
		for (int i = 0; i < list.size(); i++) {
            System.out.print(list.get(i).getKey() + " ");
        }     
	}

}