字节跳动 笔试最后一题,请各位大佬看看对吗?
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Scanner;
public class Test12 {
public static void main(String[] args) {
Scanner s = new Scanner(System.in);
int totalPcounts = s.nextInt();
int doubleCouts = s.nextInt();
int a[] = new int[doubleCouts*2];
Map<Integer,Integer> countMap = new HashMap<Integer, Integer>();
Map<Integer,Integer> countMap2 = new HashMap<Integer, Integer>();
for(int i =0; i <doubleCouts*2;i++){
a[i] = s.nextInt();
}
for(int j =1; j<a.length;j=j+2){
countMap.put(a[j], a[j-1]);
countMap2.put(a[j], 1);
if(countMap.containsKey(a[j-1])){
Integer value = countMap2.get(a[j-1]);
if(value != a[j])
countMap2.put(a[j],value+1 );
}
}
int res = 0;
for(int q=1; q<=totalPcounts;q++){
if(countMap2.get(q)==totalPcounts-1){
res++;
}
}
System.out.println(res);
}
}
#笔试题目##字节跳动#import java.util.Iterator;
import java.util.Map;
import java.util.Scanner;
public class Test12 {
public static void main(String[] args) {
Scanner s = new Scanner(System.in);
int totalPcounts = s.nextInt();
int doubleCouts = s.nextInt();
int a[] = new int[doubleCouts*2];
Map<Integer,Integer> countMap = new HashMap<Integer, Integer>();
Map<Integer,Integer> countMap2 = new HashMap<Integer, Integer>();
for(int i =0; i <doubleCouts*2;i++){
a[i] = s.nextInt();
}
for(int j =1; j<a.length;j=j+2){
countMap.put(a[j], a[j-1]);
countMap2.put(a[j], 1);
if(countMap.containsKey(a[j-1])){
Integer value = countMap2.get(a[j-1]);
if(value != a[j])
countMap2.put(a[j],value+1 );
}
}
int res = 0;
for(int q=1; q<=totalPcounts;q++){
if(countMap2.get(q)==totalPcounts-1){
res++;
}
}
System.out.println(res);
}
}