题目描述
给定一个整型数组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);
}
}
}