Leetcode 97 交错字符串

题目

给定三个字符串 s1、s2、s3，请你帮忙验证 s3 是否是由 s1 和 s2 交错 组成的。

两个字符串 s 和 t 交错 的定义与过程如下，其中每个字符串都会被分割成若干 非空 子字符串：

s = s1 + s2 + … + sn
t = t1 + t2 + … + tm
|n - m| <= 1
交错 是 s1 + t1 + s2 + t2 + s3 + t3 + … 或者 t1 + s1 + t2 + s2 + t3 + s3 + …
提示：a + b 意味着字符串 a 和 b 连接。

解题思路

  一开始看字符串的长度才100，就想直接DFS了，但感觉没什么能剪枝的地方，用DFS会爆栈。这题看起来也不像是那种合并链表的题。后来想到了动态规划，构建一个二维数组 dp ，dp[i + 1][j + 1] 表示用 s1 里的前i个字符和 s2 里的前 j个字符能否构成 s3 的前 i + j + 1 个字符（此处的个数都是下标）。对于当前状态dp[i + 1][j + 1] ，由于每次只能选 s1 或者 s2 的下一个字符，那么它仅能由 dp[i][j + 1] 和 dp[i + 1][j] 转移得到。我们只需要判断选择的字符与 s3 对应长度的字符是否相等，结合上一个节点的状态来更新当前节点的状态即可。

代码

class Solution {
    public boolean isInterleave(String s1, String s2, String s3) {
        int length1 = s1.length(), length2 = s2.length(), length3 = s3.length();
        if (length1 + length2 != length3) return false;
        boolean dp[][] = new boolean[length1 + 1][length2 + 1];

        dp[0][0] = true;
        for (int i = 0; i < length2; i++) {
            if (s2.charAt(i) == s3.charAt(i)) dp[0][i + 1] = true;
            else break;
        }
        for (int i = 0; i < length1; i++) {
            if (s1.charAt(i) == s3.charAt(i)) dp[i + 1][0] = true;
            else break;
        }

        for (int i = 0; i < length1; i++) {
            for (int j = 0; j < length2; j++) {
                boolean up = dp[i][j + 1] && (s1.charAt(i) == s3.charAt(i + j + 1));
                boolean left = dp[i + 1][j] && (s2.charAt(j) == s3.charAt(i + j + 1));
                if (up || left) dp[i + 1][j + 1] = true;
            }
        }
        return dp[length1][length2];
    }
}