蓝桥杯 算法训练 出现次数最多的整数

题目

问题描述
　　编写一个程序，读入一组整数，这组整数是按照从小到大的顺序排列的，它们的个数N也是由用户输入的，最多不会超过20。然后程序将对这个数组进行统计，把出现次数最多的那个数组元素值打印出来。如果有两个元素值出现的次数相同，即并列第一，那么只打印比较小的那个值。
　　输入格式：第一行是一个整数N，N £ 20；接下来有N行，每一行表示一个整数，并且按照从小到大的顺序排列。
　　输出格式：输出只有一行，即出现次数最多的那个元素值。
输入输出样例
样例输入
5
100
150
150
200
250
样例输出
150

思路

用map记录次数和数字，但这题可以用数组做，因为题目输入是由小到大的。
这题有个坑，要单独判断0。如果是0的话不输出。

代码

import java.util.Iterator;
import java.util.Map;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeMap;

public class Main {
    public static void main(String[] args) {
        Scanner scan=new Scanner(System.in);
        Map<Integer, Integer> map=new TreeMap();
        int n=scan.nextInt();
        int input,output,max;
        if(n<=0) {
            return;
        }else {
        for(int i=0;i<n;i++) {
            input=scan.nextInt();
            if(!map.containsKey(input)) {
                map.put(input, 1);
            }else {
                map.put(input,map.get(input)+1);
            }
        }
        Set<Integer> keyset=map.keySet();
        Iterator<Integer> it=keyset.iterator();
        output=it.next();
        max=map.get(output);
        while(it.hasNext()) {
            input=it.next();
            if(map.get(input)>max) {
                output=input;
                max=map.get(input);
            }
        }
        System.out.println(output);
        }
    }  
    
}