1. 程式人生 > 其它 >【leetcode】842將陣列拆分成斐波那契數列(三剪枝)

【leetcode】842將陣列拆分成斐波那契數列(三剪枝)

技術標籤:leetcodeleetcode剪枝

要成為更好的人

題目

給定一個數字字串 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 本身。

返回從 S 拆分出來的任意一組斐波那契式的序列塊,如果不能拆分則返回 []。

示例 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] 也同樣被接受。

提示:

  1. <= S.length <= 200
  2. 字串 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; } };

這裡是出處:
大佬題解

寫不出來了,爬去看大資料了