[leetcode刷题系列]Scramble String

这题也没说数据规模, 一开始按照100来的。 跑大数据的时候TLE了。 于是怀疑是不是每次初始化花太多时间了

所以就改成了50就过了- -,大概是n^4的算法, 有点暴力了。 不知道有没有其他更好的。当然说是n^4,实际执行效率肯定

远好于n^4的估算。


const int MAXN = 50 + 5;

string s1, s2;
int dp[MAXN][MAXN][MAXN];

int get(int st1, int st2, int len){
    int& ret = dp[st1][st2][len];
    if(ret != -1)
        return ret;
    if(len == 1)
        return ret = s1[st1] == s2[st2];
    for(int i = 1; i < len; ++ i){
        // 1
        if(get(st1, st2, i) && get(st1 + i, st2 + i, len - i))
            return ret = 1;
        if(get(st1, st2 + len - i, i) && get(st1 +  i, st2, len - i))
            return ret = 1;
    }
    return ret = 0;
}
class Solution {
public:
    bool isScramble(string s1, string s2) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        if(s1.size() != s2.size())
            return false;
        ::s1 = s1;
        ::s2 = s2;
        memset(dp, 0xff, sizeof(dp));
        return get(0, 0, s1.size());
    }
};


猜你喜欢

转载自blog.csdn.net/sigh1988/article/details/9956787