【leetcode】842將陣列拆分成斐波那契數列(三剪枝)
阿新 • • 發佈:2020-12-10
題目
給定一個數字字串 S,比如 S = “123456579”,我們可以將它分成斐波那契式的序列 [123, 456, 579]。
形式上,斐波那契式序列是一個非負整數列表 F,且滿足:
0 <= F[i] <= 2^31 - 1,(也就是說,每個整數都符合 32 位有符號整數型別);
F.length >= 3;
對於所有的0 <= i < F.length - 2,都有 F[i] + F[i+1] = F[i+2] 成立。
另外,請注意,將字串拆分成小塊時,每個塊的數字一定不要以零開頭,除非這個塊是數字 0 本身。
示例 1:
輸入:“123456579”
輸出:[123,456,579]
示例 2:
輸入: “11235813”
輸出: [1,1,2,3,5,8,13]
示例 3:
輸入: “112358130”
輸出: []
解釋: 這項任務無法完成。
示例 4:
輸入:“0123”
輸出:[]
解釋:每個塊的數字不能以零開頭,因此 “01”,“2”,“3” 不是有效答案。
示例 5:
輸入: “1101111”
輸出: [110, 1, 111]
解釋: 輸出 [11,0,11,11] 也同樣被接受。
提示:
- <= S.length <= 200
- 字串 S 中只含有數字。
思路
題目看完一如既往地懵掉了
大眼掃了一遍討論和題解,用帶回溯的搜尋和剪枝,嘗試了一下,失敗。
題外話:
好多人吐槽測試資料“539834657215398346785398346991079669377161950407626991734534318677529701785098211336528511”
xswl
程式碼
來看一下大佬的東西:
class Solution {
public:
vector<int> res;
vector<int> splitIntoFibonacci(string S) {
if ( S.empty()) return {};
vector<int> vec;
dfs(S, 0, vec, 0);
return res;
}
void dfs(string s, int str_index, vector<int>& vec, int vec_index) {
int size = s.size();
if (!res.empty()) return;
if (str_index == size && vec.size() >= 3) {//遞迴返回條件
res = vec;
return;
}
for (int i = 1; i <= size - str_index && i <= 10; ++i) {//擷取長度
string sub = s.substr(str_index, i);
long cur = _atoi(sub);
if (cur >= INT_MAX) return; // 剪枝1:拆分出的數必須符合 32 位有符號整數型別,即每個數必須在[0, 2^31−1] 的範圍內
if (vec_index < 2 && cur != -1) {//直接下一層
if (sub.size() > size - str_index - i) return;//剪枝
vec.push_back(cur);
dfs(s, str_index + i, vec, vec_index + 1);
vec.pop_back();
}
else if (cur != -1) {
long last2 = vec[vec_index - 2];//防止大數
long last1 = vec[vec_index - 1];//防止大數
long sum = last1 + last2; //防止大數
if (sum > INT_MAX) return;
if (cur == last1 + last2) {
vec.push_back(cur);
dfs(s, str_index + i, vec, vec_index + 1);
vec.pop_back();
}
else if (cur > last1 + last2) return;//剪枝2: 如果列表中至少有 22 個數,並且拆分出的數已經大於最後 22 個數的和,就不需要繼續嘗試拆分了。
else continue;
}
else return;// 剪枝3 :0開頭直接返回
}
}
int _atoi(string str) {
long sum = 0; //防止大數
int size = str.size();
if (size > 1 && str[0] == '0') return -1;//0開頭不符合條件
for (int i = 0; i < str.size(); ++i) {
sum *= 10;
sum += (str[i] - '0');
}
return sum;
}
};
這裡是出處:
大佬題解
寫不出來了,爬去看大資料了