LC 3388. Count Beautiful Splits in an Array

https://leetcode.com/problems/count-beautiful-splits-in-an-array/description/

Tag: Hash

typedef long long ll;
class Solution {
public:
    vector<ll> preh, preb;
    ll mod = 1e9 + 7;
    ll base = 31;
    void prepare(vector<int> &nums) {
        int n = nums.size();
        preh.resize(n + 1, 0);
        preb.resize(n + 1, 1);
        for (int i = 1; i <= n; i++) {
            preh[i] = (preh[i - 1] * base + nums[i - 1]) % mod;
            preb[i] = (preb[i - 1] * base) % mod;
        }
    }
    ll calHash(int l, int r) {
        ll hash = preh[r + 1] - (preh[l] * preb[r - l + 1]) % mod;
        hash = (hash + mod) % mod;
        return hash;
    }
    bool compare(vector<int> &nums, int start1, int start2, int len) {
        return calHash(start1, start1 + len - 1) == calHash(start2, start2 + len - 1);
    }
    int beautifulSplits(vector<int>& nums) {
        prepare(nums);
        int n = nums.size();
        int ans = 0;
        for (int i = 1; i < n - 1; i++) {
            for (int j = i + 1; j < n; j++) {
                int len1 = i;
                int len2 = j - i;
                int len3 = n - j;
                bool ans1 = false, ans2 = false;
                if (len1 <= len2 && compare(nums, 0, i, len1)) ans1 = true;
                if (len2 <= len3 && compare(nums, i, j, len2)) ans2 = true;
                if (ans1 || ans2) ans += 1;
            }
        }
        return ans;
    }
};

透過 Hash 就可以知道某個範圍內的值是多少，然後直接進行比對就可以。

另一種解法是透過 z-function 找到相同的 prefix ，但是因為相關的解答都有點太過於複雜所以我沒有採用這個方法。

以下連結是關於 z-function 的，如果之後忘記可以在看這個影片補一下 z-function 的知識。