2

我的朋友给了我以下问题：

Input: A matrix of letters and a word.
Output: The frequency of the word in the matrix assuming
  you can move left, right, up and down in the matrix to form the word.

例如：

Input:
S E X Y
A S E A
A A X A
A A Y A
And word is SEXY.

Output:
4 (four times in matrix of letters)

这是我解决问题的代码：

package backtracking;

public class CountFrequency {
    private char[][] matrixOfLetter;
    private String word;
    private int n, m;
    private int lengthOfWord;
    private int[][] matrixCountFrequency;

    public CountFrequency(int n, int m, String word) {
        matrixOfLetter = new char[n][m];
        this.word = word;
        this.n = n;
        this.m = m;
        this.lengthOfWord = word.length();

        matrixCountFrequency = new int[n][m];
        for (int i = 0; i < n; ++i)
            for (int j = 0; j < n; ++j)
                matrixCountFrequency[i][j] = 0;
    }

    public static void main(String[] args) {
        CountFrequency countFrequency = new CountFrequency(4, 4, "SEXY");

        countFrequency.addMatrixOfLetter(0, 0, 'S');
        countFrequency.addMatrixOfLetter(0, 1, 'E');
        countFrequency.addMatrixOfLetter(0, 2, 'X');
        countFrequency.addMatrixOfLetter(0, 3, 'Y');
        countFrequency.addMatrixOfLetter(1, 0, 'A');
        countFrequency.addMatrixOfLetter(1, 1, 'S');
        countFrequency.addMatrixOfLetter(1, 2, 'E');
        countFrequency.addMatrixOfLetter(1, 3, 'A');
        countFrequency.addMatrixOfLetter(2, 0, 'A');
        countFrequency.addMatrixOfLetter(2, 1, 'A');
        countFrequency.addMatrixOfLetter(2, 2, 'X');
        countFrequency.addMatrixOfLetter(2, 3, 'A');
        countFrequency.addMatrixOfLetter(3, 0, 'A');
        countFrequency.addMatrixOfLetter(3, 1, 'A');
        countFrequency.addMatrixOfLetter(3, 2, 'Y');
        countFrequency.addMatrixOfLetter(3, 3, 'A');

        countFrequency.process();
        countFrequency.printResult();
    }

    public void addMatrixOfLetter(int i, int j, char c) {
        matrixOfLetter[i][j] = c;
    }

    public void process() {
        for (int i = 0; i < n; ++i)
            for (int j = 0; j < m; ++j) {
                if (word.indexOf(matrixOfLetter[i][j]) == -1) {
                    matrixCountFrequency[i][j] = -1;
                    continue;
                }
                if (matrixOfLetter[i][j] == word.charAt(lengthOfWord - 1))
                    processWithLastChar(lengthOfWord - 1, i, j);
            }
    }

    public void processWithLastChar(int indexOfWord, int row, int col) {
        matrixCountFrequency[row][col] += 1;
        if (indexOfWord == 0)
            return;
        else {
            if (row - 1 >= 0) {
                if (matrixOfLetter[row - 1][col] == word
                        .charAt(indexOfWord - 1))
                    processWithLastChar(indexOfWord - 1, row - 1, col);
            }

            if (row + 1 < lengthOfWord) {
                if (matrixOfLetter[row + 1][col] == word
                        .charAt(indexOfWord - 1))
                    processWithLastChar(indexOfWord - 1, row + 1, col);
            }

            if (col - 1 >= 0) {
                if (matrixOfLetter[row][col - 1] == word
                        .charAt(indexOfWord - 1))
                    processWithLastChar(indexOfWord - 1, row, col - 1);
            }

            if (col + 1 < lengthOfWord) {
                if (matrixOfLetter[row][col + 1] == word
                        .charAt(indexOfWord - 1))
                    processWithLastChar(indexOfWord - 1, row, col + 1);
            }
        }
    }

    public void printResult() {
        int count = 0;
        for (int i = 0; i < n; ++i)
            for (int j = 0; j < m; ++j) {
                if (word.charAt(0) == matrixOfLetter[i][j])
                    count += matrixCountFrequency[i][j];
            }

        System.out.println("Frequency is : " + count);
    }
}

我使用了回溯算法，但我只在看到单词的最后一个字母时回溯，当看到单词中最右边的字母时再次回溯。

我使用计数器矩阵来计算字母的计数频率。

这个问题可以通过动态规划算法解决吗？

还是有更好的主意？

4

1 回答 1

4

可以用动态规划来解决，我认为这是最容易理解的解决方案。

为您创建一个平行的 3 维矩阵。如果字母矩阵的尺寸为nxm并且您搜索的单词是L长字母，则您创建矩阵dp[n][m][L]。在dp[i][j][k]你的存储中，你找到了多少种方法来使用这个字母initial[i][j]作为k你单词的第一个字母。

你有dp[i][j][k] = sum(dp[i+delta1][j + delta2][k + 1])，在哪里{delta1, delta2} in {{0, 1},{0, -1}, {1, 0}, {-1, 0}}。递归的底部是delta[i][j][L - 1] = (initial[i][j] == word[L - 1])。

dp[i][j][l - 1]如果你对所有可能的 i 和 j求和，就会给出最终结果。

希望这对您有所帮助。

编辑

我不得不承认我在最初的解决方案中做了一个愚蠢的提议。我提出的动态解决方案没有考虑到我使用了哪些字母。因此对于矩阵

XXXX
XABX
XXXX

字符串 ABAB 我的算法将返回一个计数 - 从 A 开始到 B 然后返回到 A 再返回到 B。这对于您的需要可能是错误的。

遗憾地跟踪您已经访问过的内容在动态方法中并不简单，现在我开始认为回溯更适合您的问题。

顺便说一句，您在解决方案中也没有考虑到这一点，但是跟踪您在回溯期间访问的内容要容易得多。我还认为回溯在内存和性能方面会更有效。

于 2013-05-09T11:26:34.133 回答