flag
软件学院大三党,每天一道算法题,第十九天
题目介绍
你正在和你的朋友玩 猜数字(Bulls and Cows)游戏:你写下一个数字让你的朋友猜。每次他猜测后,你给他一个提示,告诉他有多少位数字和确切位置都猜对了(称为“Bulls”, 公牛),有多少位数字猜对了但是位置不对(称为“Cows”, 奶牛)。你的朋友将会根据提示继续猜,直到猜出秘密数字。
请写出一个根据秘密数字和朋友的猜测数返回提示的函数,用 A 表示公牛,用 B 表示奶牛。
请注意秘密数字和朋友的猜测数都可能含有重复数字。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/bulls-and-cows
思路
先遍历两个字符串,如果两个字符串当前索引字符相同,则将"公牛数"加一;这样公牛数很容易得到了。
接下来是母牛数:
如果两个字符串当前索引字符不同,将秘密字符串的字符放入HashMap中,进行次数统计,将猜的字符放入一个Array中。
第一次遍历完毕后进行第二次遍历,此时的array中是猜的字符串中没有对上号的字符,依次获取array中的字符,看HashMap中是否有该字符,如果有,则让"母牛数"加一,然后将该字符的次数减一,如果该字符在Map中的次数为零则清除。
关键代码
public static String getHint(String secret, String guess){
int bulls=0;//公牛数,数字和确切位置都猜对了
int cows=0;//母牛数,数字猜对了但是位置不对
ArrayList<Character> arrayList=new ArrayList();//存储guess的字符
HashMap<Character,Integer> hashMap=new HashMap<>();//存储secret的字符和数量
for(int i=0; i<secret.length();i++){
char temp=secret.charAt(i);
if(temp==guess.charAt(i))
bulls++;//A的值增加
else {
arrayList.add(guess.charAt(i));
if(hashMap.containsKey(temp))
hashMap.put(temp,hashMap.get(temp)+1);//数目增加
else
hashMap.put(temp,1);
}
}
for(Character c:arrayList){
if(hashMap.containsKey(c)){
cows++;//B的值增加
hashMap.put(c,hashMap.get(c)-1);//数目减少
if(hashMap.get(c)==0)//数目减为0,删去
hashMap.remove(c);
}
}
return bulls+"A"+cows+"B";
}