CCF 202104-4 校门外的树 Python

202104-4 校门外的树

题目链接

http://118.190.20.162/view.page?gpid=T125

题型分析

典型的DP问题

递推公式:
f[i] = f[i] + f[j] *calc(i,j)

含义:
aj - ai 表示到i为止的最后一个区间,其长度为d,该区间可以被分成以k为单位的区间。
k的取值范围是q[d] (小于d的所有约数)
f[i] 表示到第i个障碍物前已有的方案数
calc(j,i) 表示第j个障碍物到第i个障碍物之间的方案数

初始化:
f[0] = 1

代码思路:
首先求出1-M之间每个数的约数,将其存储在q中备用。
从第二个障碍物起开始计算,由于方案不能重复,因此需要设置一个st的布尔数组用于记录哪些k已经使用,注意!每一轮都要重新初始化st。
从后往前遍历j。
对于每一个j,需要计算与i的距离d,针对距离d中可以选取的所有k进行遍历,如果该k没有被访问,res+1并将其对应的st[k]=False。
此处需要注意st[d]要设置成True!!【这里我也很疑惑,欢迎大佬评论区解答】
接下来利用递推公式计算f[i]

f[n-1]即为最终方案数!

代码

# 校门外的树
n = int(input())
a = list(map(int, input().split()))

N = n+1
M = a[-1]+1

MOD = 1e9 + 7

f = [0 for i in range(N)]  # 存储状态
q = [[] for i in range(M)]  # 存储每个数的约数[因数]
# 求1-M之间每个数的约数
for i in range(1, M):
    j = i * 2
    while j < M:
        q[j].append(i)
        j += i
f[0] = 1
for i in range(1, n):
    st = [False for i in range(M)]  # bool数组  用于存储该k是否被用过 每次初始化一下st数组
    for j in range(i - 1, -1, -1):
        d = a[i] - a[j]  # 计算一下a[i]到a[j]的距离
        res = 0
        # 枚举一下d的所有约数
 
        for k in q[d]:
            if st[k]!=True:
                res += 1
                st[k] = True
        st[d] = True
        f[i] = (f[i] + f[j] * res) % MOD
print(int(f[n - 1]))

但运行超时,只能拿到60分,欢迎各位提问和改进!

上一篇:[Java] 202104-2


下一篇:202104-1 灰度直方图