入坑codewars第16天-Transformation of a Number Through Prime Factorization

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sinat_37341950/article/details/85017337

题目:

Every natural number, n, may have a prime factorization like:

We define the geometric derivative of n, as a number with the following value:

For example: calculate the value of n* for n = 24500.

24500 = 2²5³7²
n* = (2*2) * (3*5²) * (2*7) = 4200

Make a function, f that can perform this calculation

f(n) ----> n*

Every prime number will have n* = 1.

Every number that does not have an exponent ki, higher than 1, will give a n* = 1, too

f(24500) == 4200

f(997) == 1

Do your best!

题意:

题意就是按照那些公式算;首先把一个数分解成质因数相乘;然后按照第二个公式计算即可。

思路:

思路就是先分解成质因数相乘,然后保存在列表里;
然后利用列表的计数函数,计算出每个质因子的个数就是kn;
按照公式求就ok了

第一次提交的代码如下超时了:

import math
from collections import Counter
def f(n):
    m=n
    nn=1
    list1=[]
    for i in range(2,m):
        while m%i==0:
            m=m/i
            list1.append(i)
    result = Counter(list1)
    for key in result:
        nn=nn*(result[key]*(key**(result[key]-1)))
    if m==n:return 1
    return nn

我之前写过类似的知道超时原因是什么? 就是这个m太大会导致超时,解决办法就是处理一下m变成根号m:

import math
from collections import Counter
def f(n):
    m=n
    nn=1
    list1=[]
    for i in range(2,int(math.sqrt(m))+1):
        while m%i==0:
            m=m/i
            list1.append(i)
    result = Counter(list1)
    for key in result:
        nn=nn*(result[key]*(key**(result[key]-1)))
    if m==n:return 1
    return nn

然后就可以了。 

猜你喜欢

转载自blog.csdn.net/sinat_37341950/article/details/85017337
今日推荐