題目

Given s1, s2, s3, find whether s3 is formed by the interleaving of s1 and s2.

Example 1:

Input: s1 = "aabcc", s2 = "dbbca", s3 = "aadbbcbcac"
Output: true

Example 2:

Input: s1 = "aabcc", s2 = "dbbca", s3 = "aadbbbaccc"
Output: false

思路與解法

這道題的題意比較容易理解，就是讓我們判斷s1和s2是否可以交叉組合得到s3。也即從s3中找到子序列s1，並且剩下的子序列構成s2（字元相對順序不改變）。 這道題目可以利用動態規劃的思想來解決。首先，我們定義以下資料：dp[i][j]

# 注意s1[i-1]表示s1的第i個字元；s3[i+j-1]表示s3的第i+j個字元；s2[j-1]表示s2的第j個字元
dp[i][j] = dp[i-1][j] && s1[i-1] == s3[i+j-1] || dp[i][j-1] && s2[j-1] == s3[i+j-1]

程式碼實現

func isInterleave(s1 string, s2 string, s3 string) bool {
    if s1 ==
 
 "" && s2 == "" && s3 == "" {
        return true
    }
    s1Len, s2Len, s3Len := len(s1), len(s2), len(s3)
    if s1Len + s2Len != s3Len {
        return false
    }
    // 生成二維陣列(s1Len+1)*(s2Len+1)
    dp := make([][]bool, 0)
    for i:=0; i<=s1Len; i++ {
        slice := make([]bool
 
, s2Len+1)
        dp = append(dp, slice)
    }
    // 初始化，此過程比較重要，因為這是遞推的起點
    // 當j==0時，定義dp[i][0]
    for i:=1; i<=s1Len; i++ {
        if s1[i-1] == s3[i-1] {
            dp[i][0] = true
        } else {
            break
        }
    }
    // 當i==0時，定義dp[0][j]
    for i:=1; i<=s2Len; i++ {
        if s2[i-1] == s3[i-1] {
            dp[0][i] = true
        } else {
            break
        }
    }
	// 進行遞推，獲得dp[s1Len][s2Len]，注意下標
    for i:=1; i<=s1Len; i++ {
        for j:=1; j<=s2Len; j++ {
            if (dp[i-1][j] && s1[i-1] == s3[i+j-1]) || (dp[i][j-1] && s2[j-1] == s3[i+j-1]) {
                dp[i][j] = true
            }
        }
    }

    return dp[s1Len][s2Len]
}

測試結果

注意，當輸入s1，s2，s3都為空值時，我們的遞推關係不能得到正確的結果，返回的dp[0][0]預設為false。所以，在此我加了一步特判。