leetcode842. 将数组拆分成斐波那契序列(回溯)

给定一个数字字符串 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]

代码

class Solution {
    public List<Integer> splitIntoFibonacci(String S) {

        split(S,0,new ArrayList<>());
        return total==null?new ArrayList<>():total;
    }
    ArrayList<Integer> total;
    boolean flag=false;
    public void split(String S,int pos,List<Integer> list) {

        if(flag) return;//已经找到了
        if(list.size()>2&&pos==S.length())//遍历完了
        {
            flag=true;
            total=new ArrayList<>(list);
            return;
        }

        for(int len=1;len+pos<=S.length()&&len<11;len++)
        {
            String sub=S.substring(pos,pos+len);
            if(len>1&&sub.charAt(0)=='0') break;//第一位是0
         long temp=Long.parseLong(sub);
            if(temp>Integer.MAX_VALUE) break;//已经溢出了
                
            if(list.size()<2||list.get(list.size()-1)+list.get(list.size()-2)==temp)//满足条件
            {
                list.add((int)temp);
                split(S,pos+len,list);
                list.remove(list.size()-1);//回溯

            }

        }
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_44560620/article/details/107750833
今日推荐