Golang | LeetCode

Задача: 842. Split Array into Fibonacci Sequence
Сложность: medium

Вам дана строка цифр num, такая как "123456579". Мы можем разделить её на последовательность, похожую на Фибоначчи [123, 456, 579].
Формально, последовательность, похожая на Фибоначчи, это список f неотрицательных целых чисел, таких что:
0 <= f[i] < 2^31 (то есть каждое число помещается в 32-битный знаковый целый тип),
f.length >= 3, и
f[i] + f[i + 1] == f[i + 2] для всех 0 <= i < f.length - 2.
Обратите внимание, что при разделении строки на части каждая часть не должна иметь лишних ведущих нулей, за исключением случая, если эта часть является числом 0.

Верните любую последовательность, похожую на Фибоначчи, из строки num, или верните [] если это невозможно.

Пример:

Input: num = "1101111"
Output: [11,0,11,11]
Explanation: The output [110, 1, 111] would also be accepted.

👨‍💻

Алгоритм:

1⃣Переберите все возможные начальные элементы первой и второй части последовательности, проверяя, чтобы не было ведущих нулей.

2⃣Для каждой пары начальных элементов проверяйте, можно ли продолжить последовательность Фибоначчи, создавая следующую часть, которая должна быть суммой двух предыдущих частей.

3⃣Если последовательность Фибоначчи найдена, верните её, иначе продолжайте перебор.

😎

Решение:

import (
    "strconv"
)

func splitIntoFibonacci(S string) []int {
    N := len(S)
    for i := 0; i < min(10, N); i++ {
        if S[0] == '0' && i > 0 {
            break
        }
        a, _ := strconv.ParseInt(S[:i+1], 10, 64)
        if a >= (1 << 31) - 1 {
            break
        }

        for j := i + 1; j < min(i + 10, N); j++ {
            if S[i+1] == '0' && j > i+1 {
                break
            }
            b, _ := strconv.ParseInt(S[i+1:j+1], 10, 64)
            if b >= (1 << 31) - 1 {
                break
            }

            fib := []int{int(a), int(b)}
            k := j + 1
            for k < N {
                nxt := int64(fib[len(fib)-2] + fib[len(fib)-1])
                if nxt >= (1 << 31) - 1 {
                    break
                }
                nxtS := strconv.FormatInt(nxt, 10)
                if len(S[k:]) >= len(nxtS) && S[k:k+len(nxtS)] == nxtS {
                    k += len(nxtS)
                    fib = append(fib, int(nxt))
                } else {
                    break
                }
            }
            if len(fib) >= 3 {
                return fib
            }
        }
    }
    return []int{}
}

func min(a, b int) int {
    if a < b {
        return a
    }
    return b
}

Ставь 👍 и забирай 📚 Базу знаний

Please open Telegram to view this post