leetcode3351 好子序列的元素之和

给定数组num[n]，如果一个子序列中任意两个相邻元素的绝对差恰好为1，则称它为好子序列，返回nums中所有好子序列的元素之和，结果对1E9+7取模。注意，长度为1的子序列算好子序列。
1<=n<=1E5; 0<=nums[i]<=1E5

分析：设f[x]表示以x结尾的所有子序列元素之和，g[x]表示以x结尾的子序列个数，从左到右逐个元素u分析，有3种情况：
（1）u自己构成一个新的子序列：f[u]+=u，g[u]+=1；
（2）u拼到u-1后面：f[u]+=g[u-1]*u+f[u-1]，g[u]+=g[u-1]；
（3）u拼到u+1后面：f[u]+=g[u+1]*u+f[u+1]，g[u]+=g[u+1]；

// mint模板...
class Solution {
public:
    int sumOfGoodSubsequences(vector<int>& nums) {
        std::map<int,mint> f, g;
        for (auto u : nums) {
            f[u] += u;
            g[u] += 1;
            
            f[u] += f[u - 1] + u * g[u - 1];
            g[u] += g[u - 1];
            
            f[u] += f[u + 1] + u * g[u + 1];
            g[u] += g[u + 1];
        }
        mint ans = 0;
        for (auto [k,v] : f) {
            ans += v;
        }
        return ans.val();
    }
};