笔试题记录——矩阵构造、长城数组

京东笔试编程题

1、输入整数n，输出一个正方形矩阵，要求包含的数为1-n^2，并且相邻的数和为奇数(满足要求即可，答案不唯一)

解题思路：如果n为奇数，那么从左上角依次输出1,2,…,n^2，写完一行后换行即可；
如果n为偶数，则在依次输出的基础上，奇数行的每个元素向右移动一格，比如4*4的矩阵：
1 2 3 4
8 5 6 7 //先输出最大的数，然后按顺序输出
9 10 11 12
16 13 14 15

import java.util.*;
public class Main{
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int[][] res = new int[n][n];
		int num = 1;
		if(n%2 != 0) {
			for(int i=0;i<n;i++) {
				for(int j=0;j<n;j++) {
					res[i][j] = num++;
					System.out.print(res[i][j]+" ");
				}
				System.out.println();
			}
		}else {
			for(int i=0;i<n;i++) {
				for(int j=0;j<n;j++) {
					if(i%2 != 0) {
						if(j==0)
							res[i][j] = num+n-1;
						else {
							res[i][j] = num++;
							if(j==n-1)
								num++;
						}
					}
					else {
						res[i][j] = num++;
					}
					System.out.print(res[i][j]+" ");
				}
				System.out.println();
			}
		}
	}
}

2、长城数组，输入n和一个长度为n的数组，输出将一个数组变为长城的最小操作次数，每次操作可以把数组一个元素变成另一个元素的大小。长城要求数的两边都相等且和本身不相等。例如[1,4,1,4,1,4,1]

解题思路：长城即奇数位数字全部相等、偶数位数字全部相等，且奇数偶数位不全相等的数组，因此需要分别记下原数组中奇数位、偶数位上每个数字出现的次数，最终我们留下出现最多的数，其他都替换成这个数，但是有一种特殊情况就是两个数相同，此时只能让奇数位或者偶数位替换成出现第二多的数。

import java.util.*;
public class Main{
	
	public static Integer getMax(HashMap<Integer,Integer> map) {
		int max=0;
		Integer max_index = null;
		for(Integer key:map.keySet()) {
			if(map.get(key)>max) {
				max = map.get(key);
				max_index = key;
			}
		}
		return max_index;
	}
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = Integer.parseInt(sc.nextLine());
		String[] str = sc.nextLine().split(" ");
		int[] arr = new int[n];
		for(int i=0;i<n;i++) {
			arr[i] = Integer.parseInt(str[i]);
		}
		
		HashMap<Integer,Integer> odd = new HashMap<>();
		HashMap<Integer,Integer> even = new HashMap<>();
		
		for(int i=0;i<n;i++) {
			if(i%2==0) {
				if(!even.containsKey(arr[i]))
					even.put(arr[i], 1);
				else
					even.put(arr[i], even.get(arr[i])+1);
			}else {
				if(!odd.containsKey(arr[i]))
					odd.put(arr[i], 1);
				else
					odd.put(arr[i], odd.get(arr[i])+1);
			}
		}
		int odd_num = getMax(odd);
		int even_num = getMax(even);
		//如果奇数位偶数位不同，直接返回n-奇数位数字最大次数-偶数位数字最大次数
		if(odd_num != even_num) {
			System.out.println(n-odd.get(odd_num)-even.get(even_num));
		}else {
			//长城数组不能全部相同，需要找到第二高的数字
			int a = odd.get(odd_num);
			int b = even.get(even_num);
			odd.put(odd_num, 0);
			even.put(even_num, 0);
			int odd_num_2 = getMax(odd);
			int even_num_2 = getMax(even);
			if((a-odd.get(odd_num_2))>(b-even.get(even_num_2)))
				System.out.println(n-a-even.get(even_num_2));
			else
				System.out.println(n-b-odd.get(odd_num_2));
		}
	}
}