蓝桥杯2023年第十四届省赛真题-子树的大小——python（简单直白）

解题思路:
    只维护子树的最左和最右的孩子，一层一层寻找。
    如果下一层是满的，则加上该层结点总数m*（遍历层数）
    如果下一层不是满的，加上该层剩余节点总数，m-l+1
注意事项:
    构建树的话应该会超时，大家可以尝试试试。
参考代码:
## 子树的大小
## 询问次数
T=int(input())
## 输入T组共 n节点  m叉数  k个节点拥有节点数
nums=[list(map(int,input().strip().split())) for _ in range(T)]
def find_num(nums):
    count=[0]*len(nums)
    for i in range(len(nums)):
        n,m,k=nums[i][0],nums[i][1],nums[i][2]
        l,r,res=k,k,1
        while True:
            count[i]+=res
            if r*m+1>n:
                if (l-1)*m+2<=n:
                    count[i]+=n-((l-1)*m+2)+1
                break
            res*=m
            r=r*m+1
            l=(l-1)*m+2
    return count
count=find_num(nums)
for i in count:
    print(i)