程序员代码面试指南刷题--第四章.字符串的交错组成

题目描述
给定三个字符串str1、str2 和aim,如果aim包含且仅包含来自str1和str2的所有字符，而且在aim中属于str1的字符之间保持原来在str1中的顺序属于str2的字符之间保持原来在str2中的顺序，那么称aim是str1和str2的交错组成。实现一个函数，判断aim是否是str1和str2交错组成，如果是请输出“YES”，否则输出“NO”。

输入描述:

输出三行，分别表示三个字符串str1，str2和aim。

输出描述:

输出“YES”或者“NO”。（不包含引号）

示例1

输入

AB
12
1AB2

输出

YES

示例2

输入

2019
9102
22001199

输出

NO

解法一：动态规划

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args)throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String s1 = br.readLine().trim();
        String s2 = br.readLine().trim();
        String aim = br.readLine().trim();
        boolean res = getRes(s1,s2,aim);
        System.out.println(res?"YES":"NO");
    }
    public static boolean getRes(String s1,String s2,String aim){
        if(s1==null||s2==null||aim==null) return false;
        if(aim.length()!=s1.length()+s2.length()) return false;
        char[] a1 = s1.toCharArray();
        char[] a2 = s2.toCharArray();
        char[] arr = aim.toCharArray();
        boolean[][] dp = getdp(a1,a2,arr);
        return dp[a1.length][a2.length];
    }
    public static boolean[][] getdp(char[] a1,char[] a2,char[] aim){
        int len1 = a1.length;
        int len2 = a2.length;
        boolean[][] dp = new boolean[len1+1][len2+1];
        dp[0][0] = true;
        for(int i=1;i<=len2;i++){
            dp[0][i] = aim[i-1]==a2[i-1]&&dp[0][i-1]?true:false;
        }
        for(int i=1;i<=len1;i++){
            dp[i][0] = aim[i-1]==a1[i-1]&&dp[i-1][0]?true:false;
        }
        for(int i=1;i<=len1;i++){
            for(int j=1;j<=len2;j++){
                if(dp[i-1][j]){
                    dp[i][j] = aim[i+j-1]==a1[i-1]?true:false;
                }
                if(dp[i][j-1]){
                    dp[i][j] = dp[i][j]||aim[i+j-1]==a2[j-1]?true:false;
                }
            }
        }
        return dp;
    }
}

解法二：压缩空间