蓝桥杯-子树的大小

"""
题目来源:
https://www.lanqiao.cn/problems/3526/learning/?page=1&first_category_id=1&name=%E5%AD%90%E6%A0%91%E7%9A%84%E5%A4%A7%E5%B0%8F
"""

"""第一种算法思路: dfs深度优先, 通过50%, 虽然没有通过但是思路没有问题
import os
import sys

# 请在此输入您的代码
# 求有n个节点的m叉树第k个节点所在子树的节点个数
def dfs(n, m, k):
  # 子树节点个数包含根节点, 初始就有一个节点
  ans = 1
  # 以k为根节点下的m个子节点为m(k - 1) + 2 ~ m(k - 1) + m + 1
  for i in range(m * (k - 1) + 2, m * (k - 1) + m + 1 + 1):
    if i > n:
      break 
    ans += dfs(n, m, i)
  return ans

T = int(input())
for _ in range(T):
  n, m, k = map(int, input().split())
  print(dfs(n, m, k))

"""

import os
import sys

T = int(input())

for _ in range(T):
    # 以k为根节点下的m个子节点为m(k - 1) + 2 ~ m(k - 1) + m + 1
    n, m, k = map(int, input().split())
    # 表示以k为根节点下的最左边的子节点
    l = m * (k - 1) + 2
    # 表示以k为根节点下的最右边的子节点
    r = m * (k - 1) + m + 1
    # 子树的结点总数, 根节点个数为1
    ans = 1
    # 记录每一层子结点的数量
    res = 1
    # 如果子节点最右子节点小于等于n, 说明没到尽头, 这一层子节点是满的
    while r <= n:
        # 当前层子节点数目
        res = res * m
        # 更新下一层的最左边子节点      
        l = m * (l - 1) + 2 
        # 更新下一层的最右边子节点  
        r = m * (r - 1) + m + 1  
        ans += res
    ans += max(0, n - l + 1)  # 最后一层最右端点就是n, 直接用n-l+1计算出这层的结点数量
    print(ans)