题解 | #牛群定位系统#

牛群定位系统

https://www.nowcoder.com/practice/d808aaa4c13c4d38bff333aa796e6a1e?tpId=354&tqId=10594633&ru=/exam/oj&qru=/ta/interview-202-top/question-ranking&sourceUrl=%2Fexam%2Foj

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param board char字符型二维数组
     * @param words string字符串一维数组
     * @return string字符串一维数组
     */
    public String[] findWords (char[][] board, String[] words) {
        // write code here
        ArrayList<String> array = new ArrayList<String>();
        for (int x = 0; x < words.length; x++) {
            if (findWords(board, words[x], 0, 0, 0)) {
                array.add(words[x]);
            }
        }
        int len = array.size();
        String[] res = new String[len];
        for (int i = 0; i < len; i++) {
            res[i] = array.get(i);
        }
        return res;
    }
    public boolean findWords(char[][] board, String word, int i, int j, int index) {
        if (i < 0  || i >= board.length || j < 0 || j >= board[i].length) {
            return index == word.length() ? true : false;
        }
        if (index == word.length()) {
            return true;
        }

        if (board[i][j] == word.charAt(index)) {
            // 做出选择
            char c = board[i][j];
            board[i][j] = '#';
            // 四个方向寻找
            boolean up = findWords(board, word, i + 1, j, index + 1);
            boolean down = findWords(board, word, i - 1, j, index + 1);
            boolean right = findWords(board, word, i, j + 1, index + 1);
            boolean left = findWords(board, word, i, j - 1, index + 1);
            // 撤销选择
            board[i][j] = c;
            return up || down || left || right;
        } else if (board[i][j] == '#') {
            return false;
        }
        if (index != 0) {
            // 说明已经使用过相邻字符了,如果还没有找到,则说明不存在
            return false;
        }
        // 下和右方向寻找
        boolean down = findWords(board, word, i + 1, j, index);
        boolean right = findWords(board, word, i, j + 1, index);
        return down || right;
    }
}

判断二维数组中是否存在字符串words[x],如果存在就添加到最后结果中,如果不存在就继续向下找

找四个方向会造成栈溢出,所以从左上角指遍历右边和下边两个方向防止无限递归造成栈溢出。

#java##回溯算法##回溯法##递归##数组移动#
全部评论

相关推荐

只写bug的程序媛:人家说一本以上,不是及以上
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务