东东对幂运算很感兴趣,在学习的过程中东东发现了一些有趣的性质: 9^3 = 27^2, 2^10 = 32^2
东东对这个性质充满了好奇,东东现在给出一个整数n,希望你能帮助他求出满足 a^b = c^d(1 ≤ a,b,c,d ≤ n)的式子有多少个。
例如当n = 2: 1^1=1^1
1^1=1^2
1^2=1^1
1^2=1^2
2^1=2^1
2^2=2^2
一共有6个满足要求的式子
输入包括一个整数n(1 ≤ n ≤ 10^6)
输出一个整数,表示满足要求的式子个数。因为答案可能很大,输出对1000000007求模的结果
2
6
def count_expressions(n):
# 创建一个字典来存储每个幂的结果及其对应的指数对
powers = {}
# 计算所有可能的 a^b 并存储在字典中
for a in range(1, n + 1):
for b in range(1, n + 1):
power = a ** b
if power not in powers:
powers[power] = []
powers[power].append((a, b))
# 计算所有可能的 c^d 并与 a^b 的结果进行比较
count = 0
for c in range(1, n + 1):
for d in range(1, n + 1):
power = c ** d
if power in powers:
count += len(powers[power])
return count