程序员代码面试指南刷题--第八章.在数组中找到出现次数大于n/k的数

题目描述
给定一个整型数组arr,再给定一个整数k,打印所有出现次数大于n/k的数,如果没有这样的数,请打印”-1“。
输入描述:

输入包含两行,第一行输入包含两个整数n和k,第二行包含n个整数,代表数组arr。

输出描述:

输出所有出现次数大于n/k的数,如果没有这样的数,请输出”-1“。

示例1

输入

7 7
1 2 3 1 2 3 4

输出

1 2 3

解法一:利用map

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args)throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] info = br.readLine().trim().split(" ");
        int n = Integer.parseInt(info[0]);
        int k = Integer.parseInt(info[1]);
        String[] ss = br.readLine().trim().split(" ");
        int[] arr = new int[n];
        for(int i=0;i<n;i++){
            arr[i] = Integer.parseInt(ss[i]);
        }
        getRes(arr,n,k);
    }
    
    public static void getRes(int[] arr,int n,int k){
        if(arr==null||arr.length==0||n/k<0) return ;
        int c = n/k;
        HashMap<Integer,Integer> hash = new HashMap<Integer,Integer>();
        StringBuilder sb = new StringBuilder();
        for(int i=0;i<n;i++){
            int tmp = hash.get(arr[i])==null?0:hash.get(arr[i]);
            hash.put(arr[i],tmp+1);
            if(tmp==c){
                sb.append(arr[i]+" ");
            }
        }
        if(sb.length()>0){
            System.out.println(sb.toString().trim());
        }else{
            System.out.println("-1");
        }
    }
} 

解法二:空间复杂度O(K)

思路:每次删除k个不同的数。

import java.io.*;
import java.util.*;
import java.util.Map.Entry;
public class Main{
    public static void main(String[] args)throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] info = br.readLine().trim().split(" ");
        int n = Integer.parseInt(info[0]);
        int k = Integer.parseInt(info[1]);
        String[] ss = br.readLine().trim().split(" ");
        int[] arr = new int[n];
        for(int i=0;i<n;i++){
            arr[i] = Integer.parseInt(ss[i]);
        }
        String res = getRes(arr,n,k);
        System.out.println(res);
    }
    //每次删除k个最后剩下的数可能就是结果
    public static String getRes(int[] arr,int n,int k){
        if(arr==null||arr.length==0||n/k<0) return "-1";
        int c = n/k;
        HashMap<Integer,Integer> cands = new HashMap<Integer,Integer>();
        StringBuilder sb = new StringBuilder();
        for(int i=0;i<n;i++){
            if(cands.containsKey(arr[i])){
                cands.put(arr[i],cands.get(arr[i])+1);
            }else{
                if(cands.size()==k-1){
                    allMinus(cands);
                }else{
                    cands.put(arr[i],1);
                }
            }
        }
        HashMap<Integer,Integer> realRes = getReal(arr,cands);
        for(Entry<Integer, Integer> set : realRes.entrySet()){
            int key = set.getKey();
            int value = set.getValue();
            if(value>(n/k)){
                sb.append(key+" ");
            }
        }
        return sb.length()==0?"-1":sb.toString().trim();
    }
    public static HashMap<Integer,Integer> getReal(int[] arr,HashMap<Integer,Integer> cands){
        HashMap<Integer,Integer> real = new HashMap<Integer,Integer>();
        for(int i=0;i<arr.length;i++){
            if(cands.containsKey(arr[i])){
                if(real.containsKey(arr[i])){
                    real.put(arr[i],real.get(arr[i])+1);
                }else{
                    real.put(arr[i],1);
                }
            }
        }
        return real;
    }
    
    public static void allMinus(HashMap<Integer,Integer> map){
        List<Integer> removeList = new ArrayList<Integer>();
        for(Map.Entry<Integer, Integer> entry:map.entrySet()) {
            int value = entry.getValue();
            entry.setValue(value-1);
            //注意这里不能直接删除!! 会出现数组越界非法访问的情况
            if(value==1) {
                removeList.add(entry.getKey());
            }
        }
         
        for(Integer i:removeList) {
            map.remove(i);
        }
    }
} 

你可能感兴趣的:(程序员代码面试指南刷题)