题目地址:
https://leetcode.com/problems/guess-the-word/
给定一个长 n n n的字符串数组 A A A,只含英文小写字母,并且每个字符串长度相等,都是 l l l。现在要猜单词,给定一个类,输入是长 l l l的字符串,输出是该字符串与要猜的单词有几个位置是正确的。总共猜 10 10 10次,设计算法去猜单词(不一定要求猜中,但要最大可能性去争取猜中)。
先存一下数组 f [ i ] [ j ] f[i][j] f[i][j],表示 A [ i ] A[i] A[i]和 A [ j ] A[j] A[j]有几个位置是相等的。那么当我们猜单词的,我们尽量选那些如果猜错了,就能删掉最多单词的那个单词,那么可以先枚举一下每个单词,有多少个单词与之有 0 , 1 , . . . , 6 0,1,...,6 0,1,...,6个位置不同,然后取最大,那么如果这个单词不是要猜的单词,能删掉的单词是最多的(如果当前猜的单词 s s s位置正确的有 k k k位,那么就筛选出 k k k位与 s s s相同的字符串作为候选即可)。代码如下:
public class Solution {
public void findSecretWord(String[] wordlist, Master master) {
int n = wordlist.length;
int[][] f = new int[n][n];
boolean[] vis = new boolean[n];
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
for (int k = 0; k < 6; k++) {
if (wordlist[i].charAt(k) == wordlist[j].charAt(k)) {
f[i][j]++;
}
}
}
}
for (int i = 0; i < 10; i++) {
int k = -1, w = Integer.MAX_VALUE;
for (int j = 0; j < n; j++) {
if (vis[j]) {
continue;
}
int t = 0;
for (int u = 0; u <= 6; u++) {
t = Math.max(t, get(j, u, n, vis, f));
}
if (w > t) {
k = j;
w = t;
}
}
int res = master.guess(wordlist[k]);
if (res == 6) {
break;
}
vis[k] = true;
for (int j = 0; j < n; j++) {
if (f[k][j] != res) {
vis[j] = true;
}
}
}
}
int get(int j, int u, int n, boolean[] vis, int[][] f) {
int res = 0;
for (int i = 0; i < n; i++) {
if (!vis[i] && f[j][i] == u) {
res++;
}
}
return res;
}
}
interface Master {
int guess(String word);
}
时空复杂度 O ( n 2 ) O(n^2) O(n2)。