题面描述
变种汉诺塔问题和普通汉诺塔问题略有不同,规则描述如下:
- 有三根柱子,在最左侧柱子上放置着若干圆盘。与传统汉诺塔不同的是,其中存在部分大小相同的圆盘。
- 要求包括初始状态在内,每个圆盘上方放置的圆盘不得大于该圆盘,即圆盘上方只能放置小于自己或和自己相同大小的圆盘。
- 每次移动只能将某柱子最顶部的一个圆盘移动到另一柱子的最顶部。
- 需要注意的是,大小相同的圆盘具有的其他特征是不一样的,例如不同颜色。
最后需要保证 2 号柱子上的圆盘排列顺序,和开始时的 0 号柱子上的顺序完全相同。
求将初态 0 号柱子上的所有圆盘全部移到 2 号柱子上最优策略的步数 l 对 m 取模后的值。
输入数据
对于每组数据:
第一行有一个整数 t (1 ≤ t ≤ 100 ) ,表示有 t 组数据。
第一行包括 2 个数字 n,m (1≤n≤15000, 1≤m≤1000000) ,其中 n 代表圆盘种类的个数;
第二行包括 n 个数字 a1, … , an (1 ≤ ai≤ 99 ),其中 ai 代表大小为 i 的圆盘个数。
输出数据
对于每组数据,输出一行,若最优策略的步数为 l ,则输出 l mod m 。
样例输入
2
2 1000
1 2
3 1000
1 2 3
样例输出
7
21
心得:惭愧…没理解这个,参考网上C++改的
T = int(input())
for t in range(T):
n, m = list(map(int, input().split()))
s = [int(item) for item in input().split()]
temp = [s[0]]
for i in range(1,n):
temp.append((2*temp[i-1]+s[i])%m)
sumN = [2*s[0]-1]
for i in range(1,n):
if s[i]==1:
sumN.append(2*temp[i-1]%m+1)
else:
sumN.append((2*temp[i-1]+2*s[i])%m+sumN[i-1])
print(sumN[n-1]%m)