[题解] Tree IV

60~70%

可以直接对整棵树进行Dfs,即从根结点1号点开始,若当前结点为,则递归搜索两个子结点,然后暴力计算答案,时间复杂度为,根据语言选择的不同可能会有不同的得分。

100%

本题利用了两个性质,同时这也是完全二叉树最常用的两个性质:

  • 若父结点为,则其两个子结点分别为
  • 每一层均有个结点

可以枚举深度,同时维护每一层结点的范围,分别记作,此时,这一层答案为,然后求和即可。

注意到此题需要对进行取模,所以我们需要预处理出的逆元,按照上述方法进行计算即可。

时间复杂度

参考代码:

class Solution {
public:
    typedef long long ll;
    const ll mod = 998244353;

    inline ll qpow(ll a, ll b) {
        ll ret = 1;
        while (b) {
            if (b & 1) ret = ret * a % mod;
            a = a * a % mod, b >>= 1;
        }
        return ret % mod;
    }

    long long tree4(long long n) {
        ll ans = 0, inv2 = qpow(2, mod - 2), l, r, dep;
        for (l = 1, r = 1, dep = 1; r <= n; ++dep) {
            ans = (ans + dep * ((l + r) % mod * (r - l + 1) % mod * inv2 % mod) % mod) % mod;
            l = r + 1, r = (r + 1) * 2 - 1;
        }
        ans = (ans + dep * ((l + n) % mod * (n - l + 1) % mod * inv2 % mod) % mod) % mod;
        return ans;
    }
};

有关逆元的使用方法,可以自行搜索学习,这里仅提供一个简单的解释:
在除法运算下,除以一个数字后进行取模,等价于乘以这个数字对模数的逆元然后再取模。

全部评论

相关推荐

投递美团等公司10个岗位
点赞 评论 收藏
转发
点赞 收藏 评论
分享
牛客网
牛客企业服务