牛客练习赛96

dongdongdong122

已于 2022-02-26 15:39:21 修改

阅读量300

点赞数

文章标签：算法图论动态规划

于 2022-02-25 21:08:18 首次发布

本文链接：https://blog.csdn.net/dongdongdong122/article/details/123141161

版权

题目描述

求一颗n层的满k叉树，求任意两点之间距离和等于多少，答案对109+7取模

树上两点距离：沿着最短路径从u走到v经过的边的数量

n层的满k叉树：叶子节点到根的距离是n−1,每个点都有k个儿子

输入描述:

一行两个正整数代表n,k

2≤n,k≤1e6

输出述:

输出一行一个数代表答案

示例1

输入

3 3

输出

题解：

分层分别统计每一条边的贡献即可，时间复杂度O(n)。

#include<bits/stdc++.h>
using namespace  std;
using ll=long long;
const ll mod=1000000000+7;
ll quick(ll a,ll b)
{
    ll ans=1;
    while(b)
    {
        if(b&1)ans=ans*a%mod;
        a=a*a%mod;
        b/=2;
    }
    return ans;
}
ll K[1000000+10];
ll KK[1000000+10];
int main()
{
    ll n,k;
    cin>>n>>k;
    ll ans=0;
    ll inv=quick(k-1,mod-2);
    ll ANS=(quick(k,n)-1)*inv%mod;//总的点数
    K[1]=k;
    KK[1]=1;
    for(int i=2; i<n; i++)
    {
        K[i]=K[i-1]*k%mod;
        KK[i]=(KK[i-1]+K[i-1])%mod;
//        cout<<K[i]<<' '<<KK[i]<<endl;
    }
    for(int i=1; i<n; i++)
        ans=(ans+K[i]*KK[n-i]%mod*(ANS-KK[n-i]+mod)%mod)%mod;
    cout<<ans<<endl;
}