在Python中查找数字的所有因子的最有效方法是什么?

时间:2011-07-23 12:00:18

标签: python algorithm python-2.7 performance factorization

有人可以向我解释一种在Python(2.7)中找到数字的所有因子的有效方法吗?

我可以创建算法来完成这项工作,但我认为它的编码很差,并且执行大量数据的结果需要很长时间。

25 个答案:

答案 0 :(得分:231)

from functools import reduce

def factors(n):    
    return set(reduce(list.__add__, 
                ([i, n//i] for i in range(1, int(n**0.5) + 1) if n % i == 0)))

这将很快返回数字n的所有因素。

为什么将平方根作为上限?

sqrt(x) * sqrt(x) = x。因此,如果两个因素相同,它们都是平方根。如果你将一个因子做大,你必须使另一个因子变小。这意味着两者中的一个将始终小于或等于sqrt(x),因此您只需要搜索到该点以找到两个匹配因子中的一个。然后,您可以使用x / fac1获取fac2

reduce(list.__add__, ...)正在使用[fac1, fac2]的小列表,并将它们连接在一个长列表中。

[i, n/i] for i in range(1, int(sqrt(n)) + 1) if n % i == 0返回一对因子,如果将n除以较小值的余数为零(它也不需要检查较大的那个;它只是通过划分得到它n由较小的一个。)

外面的set(...)正在摆脱重复,这只发生在完美的方块上。对于n = 4,这将返回2两次,因此set将删除其中一个。

答案 1 :(得分:45)

@agf提供的解决方案很棒,但通过检查奇偶校验,可以使任意奇数数字的运行时间缩短约50%。由于奇数的因子本身总是奇数,因此在处理奇数时没有必要检查它们。

我刚刚开始解决Project Euler个谜题。在某些问题中,在两个嵌套的for循环内部会调用除数检查,因此该函数的性能至关重要。

将这一事实与agf优秀的解决方案相结合,我最终得到了这个功能:

from math import sqrt
def factors(n):
        step = 2 if n%2 else 1
        return set(reduce(list.__add__,
                    ([i, n//i] for i in range(1, int(sqrt(n))+1, step) if n % i == 0)))

但是,对于较小的数字(〜<100),此更改带来的额外开销可能会导致函数花费更长时间。

我跑了一些测试来检查速度。以下是使用的代码。为了生成不同的图,我相应地更改了X = range(1,100,1)

import timeit
from math import sqrt
from matplotlib.pyplot import plot, legend, show

def factors_1(n):
    step = 2 if n%2 else 1
    return set(reduce(list.__add__,
                ([i, n//i] for i in range(1, int(sqrt(n))+1, step) if n % i == 0)))

def factors_2(n):
    return set(reduce(list.__add__,
                ([i, n//i] for i in range(1, int(sqrt(n)) + 1) if n % i == 0)))

X = range(1,100000,1000)
Y = []
for i in X:
    f_1 = timeit.timeit('factors_1({})'.format(i), setup='from __main__ import factors_1', number=10000)
    f_2 = timeit.timeit('factors_2({})'.format(i), setup='from __main__ import factors_2', number=10000)
    Y.append(f_1/f_2)
plot(X,Y, label='Running time with/without parity check')
legend()
show()

X =范围(1,100,1) X = range(1,100,1)

这里没有显着差异,但数字越大,优势显而易见:

X =范围(1,100000,1000)(仅奇数) X = range(1,100000,1000) (only odd numbers)

X =范围(2,100000,100)(仅偶数) X = range(2,100000,100) (only even numbers)

X =范围(1,100000,1001)(交替奇偶校验) X = range(1,100000,1001) (alternating parity)

答案 2 :(得分:26)

agf的回答非常酷。我想看看是否可以重写它以避免使用reduce()。这就是我想出的:

import itertools
flatten_iter = itertools.chain.from_iterable
def factors(n):
    return set(flatten_iter((i, n//i) 
                for i in range(1, int(n**0.5)+1) if n % i == 0))

我还尝试了一个使用棘手的生成器函数的版本:

def factors(n):
    return set(x for tup in ([i, n//i] 
                for i in range(1, int(n**0.5)+1) if n % i == 0) for x in tup)

我通过计算计时:

start = 10000000
end = start + 40000
for n in range(start, end):
    factors(n)

我运行一次让Python编译它,然后在time(1)命令下运行三次并保持最佳时间。

  • 缩小版:11.58秒
  • itertools版本:11.49秒
  • 棘手的版本:​​11.12秒

请注意,itertools版本正在构建一个元组并将其传递给flatten_iter()。如果我改变代码来构建列表,它会稍微减慢:

  • iterools(list)版本:11.62秒

我相信棘手的生成器函数版本在Python中是最快的。但它并不比降低版本快得多,根据我的测量值大约快4%。

答案 3 :(得分:11)

agf答案的另一种方法:

def factors(n):    
    result = set()
    for i in range(1, int(n ** 0.5) + 1):
        div, mod = divmod(n, i)
        if mod == 0:
            result |= {i, div}
    return result

答案 4 :(得分:6)

进一步改善afg&amp; eryksun的解决方案。 下面的代码返回所有因子的排序列表,而不会改变运行时渐近的复杂性:

    def factors(n):    
        l1, l2 = [], []
        for i in range(1, int(n ** 0.5) + 1):
            q,r = n//i, n%i     # Alter: divmod() fn can be used.
            if r == 0:
                l1.append(i) 
                l2.append(q)    # q's obtained are decreasing.
        if l1[-1] == l2[-1]:    # To avoid duplication of the possible factor sqrt(n)
            l1.pop()
        l2.reverse()
        return l1 + l2

想法:而不是使用list.sort()函数来获得一个排序列表,它给出了nlog(n)的复杂性;在l2上使用list.reverse()要快得多,这会带来O(n)的复杂性。 (这就是python的制作方法。) 在l2.reverse()之后,l2可以附加到l1以获得排序的因子列表。

注意,l1包含 i -s,它们正在增加。 l2包含 q -s,它们正在减少。这就是使用上述想法背后的原因。

答案 5 :(得分:6)

我已经尝试了大部分这些精彩的答案,并将其效率与我的简单功能进行比较,但我不断看到我的表现超出了此处列出的效果。我想我会分享它,看看你们都在想什么。

def factors(n):
    results = set()
    for i in xrange(1, int(math.sqrt(n)) + 1):
        if n % i == 0:
            results.add(i)
            results.add(int(n/i))
    return results

正如它所写的那样,你必须导入数学进行测试,但用n **。5替换math.sqrt(n)应该也能正常工作。我不打算浪费时间检查重复项,因为重复项不会存在于集合中。

答案 6 :(得分:6)

对于n高达10 ** 16(甚至可能更多),这是一个快速纯Python 3.6解决方案,

from itertools import compress

def primes(n):
    """ Returns  a list of primes < n for n > 2 """
    sieve = bytearray([True]) * (n//2)
    for i in range(3,int(n**0.5)+1,2):
        if sieve[i//2]:
            sieve[i*i//2::i] = bytearray((n-i*i-1)//(2*i)+1)
    return [2,*compress(range(3,n,2), sieve[1:])]

def factorization(n):
    """ Returns a list of the prime factorization of n """
    pf = []
    for p in primeslist:
      if p*p > n : break
      count = 0
      while not n % p:
        n //= p
        count += 1
      if count > 0: pf.append((p, count))
    if n > 1: pf.append((n, 1))
    return pf

def divisors(n):
    """ Returns an unsorted list of the divisors of n """
    divs = [1]
    for p, e in factorization(n):
        divs += [x*p**k for k in range(1,e+1) for x in divs]
    return divs

n = 600851475143
primeslist = primes(int(n**0.5)+1) 
print(divisors(n))

答案 7 :(得分:5)

这是另一个没有减少的替代方案,它可以很好地处理大数字。它使用sum来展平列表。

def factors(n):
    return set(sum([[i, n//i] for i in xrange(1, int(n**0.5)+1) if not n%i], []))

答案 8 :(得分:5)

这是@ agf解决方案的替代方案,它以更加pythonic的方式实现相同的算法:

def factors(n):
    return set(
        factor for i in range(1, int(n**0.5) + 1) if n % i == 0
        for factor in (i, n//i)
    )

此解决方案适用于Python 2和Python 3,没有导入,并且更具可读性。我没有测试过这种方法的性能,但渐渐地它应该是相同的,如果性能是一个严重的问题,那么这两种解决方案都不是最佳的。

答案 9 :(得分:4)

请务必抓取大于sqrt(number_to_factor)的数字,以获取异常数字,例如99,其中包含3 * 3 * 11和floor sqrt(99)+1 == 10

import math

def factor(x):
  if x == 0 or x == 1:
    return None
  res = []
  for i in range(2,int(math.floor(math.sqrt(x)+1))):
    while x % i == 0:
      x /= i
      res.append(i)
  if x != 1: # Unusual numbers
    res.append(x)
  return res

答案 10 :(得分:4)

SymPy中有一种名为factorint的行业强度算法:

>>> from sympy import factorint
>>> factorint(2**70 + 3**80) 
{5: 2,
 41: 1,
 101: 1,
 181: 1,
 821: 1,
 1597: 1,
 5393: 1,
 27188665321L: 1,
 41030818561L: 1}

这花了不到一分钟。它在一系列方法之间切换。请参阅上面链接的文档。

考虑到所有主要因素,可以轻松构建所有其他因素。

请注意,即使接受的答案被允许运行足够长的时间(即永恒)以考虑上述数字,对于某些大数字它也会失败,例如以下示例。这是由于草率int(n**0.5)。例如,n = 10000000000000079**2时,我们有

>>> int(n**0.5)
10000000000000078L

10000000000000079 is a prime开始,接受的答案的算法永远不会找到这个因素。请注意,它不仅仅是一个一个一个;对于更大的数字,它会更多。因此,最好避免使用此类算法中的浮点数。

答案 11 :(得分:2)

您的最大因数不超过您的数字,所以,假设

def factors(n):
    factors = []
    for i in range(1, n//2+1):
        if n % i == 0:
            factors.append (i)
    factors.append(n)

    return factors

瞧!

答案 12 :(得分:2)

如果您希望使用素数更快,这是一个示例。这些列表很容易在互联网上找到。我在代码中添加了注释。

# http://primes.utm.edu/lists/small/10000.txt
# First 10000 primes

_PRIMES = (2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 
        31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 
        73, 79, 83, 89, 97, 101, 103, 107, 109, 113, 
        127, 131, 137, 139, 149, 151, 157, 163, 167, 173, 
        179, 181, 191, 193, 197, 199, 211, 223, 227, 229, 
        233, 239, 241, 251, 257, 263, 269, 271, 277, 281, 
        283, 293, 307, 311, 313, 317, 331, 337, 347, 349, 
        353, 359, 367, 373, 379, 383, 389, 397, 401, 409, 
        419, 421, 431, 433, 439, 443, 449, 457, 461, 463, 
        467, 479, 487, 491, 499, 503, 509, 521, 523, 541, 
        547, 557, 563, 569, 571, 577, 587, 593, 599, 601, 
        607, 613, 617, 619, 631, 641, 643, 647, 653, 659, 
        661, 673, 677, 683, 691, 701, 709, 719, 727, 733, 
        739, 743, 751, 757, 761, 769, 773, 787, 797, 809, 
        811, 821, 823, 827, 829, 839, 853, 857, 859, 863, 
        877, 881, 883, 887, 907, 911, 919, 929, 937, 941, 
        947, 953, 967, 971, 977, 983, 991, 997, 1009, 1013, 
# Mising a lot of primes for the purpose of the example
)


from bisect import bisect_left as _bisect_left
from math import sqrt as _sqrt


def get_factors(n):
    assert isinstance(n, int), "n must be an integer."
    assert n > 0, "n must be greather than zero."
    limit = pow(_PRIMES[-1], 2)
    assert n <= limit, "n is greather then the limit of {0}".format(limit)
    result = set((1, n))
    root = int(_sqrt(n))
    primes = [t for t in get_primes_smaller_than(root + 1) if not n % t]
    result.update(primes)  # Add all the primes factors less or equal to root square
    for t in primes:
        result.update(get_factors(n/t))  # Add all the factors associted for the primes by using the same process
    return sorted(result)


def get_primes_smaller_than(n):
    return _PRIMES[:_bisect_left(_PRIMES, n)]

答案 13 :(得分:2)

一种比这里提出的算法更有效的算法(特别是如果n中存在小的素数因子)。这里的技巧是调整限制,每次找到素因子时都需要进行试验划分:

def factors(n):
    '''
    return prime factors and multiplicity of n
    n = p0^e0 * p1^e1 * ... * pk^ek encoded as
    res = [(p0, e0), (p1, e1), ..., (pk, ek)]
    '''

    res = []

    # get rid of all the factors of 2 using bit shifts
    mult = 0
    while not n & 1:
        mult += 1
        n >>= 1
    if mult != 0:
        res.append((2, mult))

    limit = round(sqrt(n))
    test_prime = 3
    while test_prime <= limit:
        mult = 0
        while n % test_prime == 0:
            mult += 1
            n //= test_prime
        if mult != 0:
            res.append((test_prime, mult))
            if n == 1:              # only useful if ek >= 3 (ek: multiplicity
                break               # of the last prime) 
            limit = round(sqrt(n))  # adjust the limit
        test_prime += 2             # will often not be prime...
    if n != 1:
        res.append((n, 1))
    return res

这当然仍然是试验分裂,而不是更加花哨。因此效率仍然非常有限(特别是对于没有小除数的大数字)。

这是python3;分区//应该是你需要适应python 2的唯一东西(添加from __future__ import division)。

答案 14 :(得分:1)

查找数字因子的最简单方法:

def factors(x):
    return [i for i in range(1,x+1) if x%i==0]

答案 15 :(得分:1)

如果你不想使用任何库,我认为这是最简单的方法

def factors(n):
    l=[] #emoty list
    # appending the factors in the list
    for i in range(1,n+1):
        if n%i==0:
            l.append(i)

我是一名中级程序员。所以以防万一,如果我错了,请纠正我

答案 16 :(得分:1)

使用set(...)会使代码稍慢,而且只有在检查平方根时才需要。这是我的版本:

def factors(num):
    if (num == 1 or num == 0):
        return []
    f = [1]
    sq = int(math.sqrt(num))
    for i in range(2, sq):
        if num % i == 0:
            f.append(i)
            f.append(num/i)
    if sq > 1 and num % sq == 0:
        f.append(sq)
        if sq*sq != num:
            f.append(num/sq)
    return f

对于像12这样的数字,if sq*sq != num:条件是必需的,其中平方根不是整数,但是平方根的底数是一个因子。

请注意,此版本本身不会返回数字,但如果您需要,则可以轻松修复。输出也没有排序。

我计划在所有数字1-200上运行10000次,在所有数字1-5000上运行100次。它优于我测试的所有其他版本,包括dansalmo,Jason Schorn,oxrock,agf,steveha和eryksun的解决方案,尽管oxrock是迄今为止最接近的版本。

答案 17 :(得分:1)

使用像下面列表理解这样简单的东西,注意我们不需要测试1和我们想要找到的数字:

def factors(n):
    return [x for x in range(2, n//2+1) if n%x == 0]

在参考平方根的使用时,假设我们想要找到10的因子。sqrt(10) = 4的整数部分因此range(1, int(sqrt(10))) = [1, 2, 3, 4]并且测试最多4个明显错过5。

除非我遗漏了一些我建议的内容,如果你必须这样做,请使用int(ceil(sqrt(x)))。当然,这会产生大量不必要的函数调用。

答案 18 :(得分:0)

循环直到在元组的 x 或 v 中找到重复项,其中 x 是分母,v 是结果。

number=30
tuple_list=[]
for i in np.arange(1,number):
    if number%i==0:
         other=int(number/i)
         if any([(x,v) for (x,v) in tuple_list if (i==x) or (i==v)])==True:
             break
         tuple_list.append((i,other))
    
 flattened = [item for sublist in tuple_list for item in sublist]              
 print(sorted(flattened))

输出

 [1, 2, 3, 5, 6, 10, 15, 30]

答案 19 :(得分:0)

我在 python 中使用 cypari 库找到了一个简单的解决方案。 这是a link

import cypari
def get_divisors(n):
   divisors = cypari.pari('divisors({})'.format(n))
   return divisors
print(get_divisors(24))

输出

[1, 2, 3, 4, 6, 8, 12, 24]

答案 20 :(得分:0)

 import math

    '''
    I applied finding prime factorization to solve this. (Trial Division)
    It's not complicated
    '''


    def generate_factors(n):
        lower_bound_check = int(math.sqrt(n))  # determine lowest bound divisor range [16 = 4]
        factors = set()  # store factors
        for divisors in range(1, lower_bound_check + 1):  # loop [1 .. 4]
            if n % divisors == 0:
                factors.add(divisors)  # lower bound divisor is found 16 [ 1, 2, 4]
                factors.add(n // divisors)  # get upper divisor from lower [ 16 / 1 = 16, 16 / 2 = 8, 16 / 4 = 4]
        return factors  # [1, 2, 4, 8 16]


    print(generate_factors(12)) # {1, 2, 3, 4, 6, 12} -> pycharm output

 Pierre Vriens hopefully this makes more sense. this is an O(nlogn) solution. 

答案 21 :(得分:0)

import 'dart:math';
generateFactorsOfN(N){
  //determine lowest bound divisor range
  final lowerBoundCheck = sqrt(N).toInt();
  var factors = Set<int>(); //stores factors
  /**
   * Lets take 16:
   * 4 = sqrt(16)
   * start from 1 ...  4 inclusive
   * check mod 16 % 1 == 0?  set[1, (16 / 1)]
   * check mod 16 % 2 == 0?  set[1, (16 / 1) , 2 , (16 / 2)]
   * check mod 16 % 3 == 0?  set[1, (16 / 1) , 2 , (16 / 2)] -> unchanged
   * check mod 16 % 4 == 0?  set[1, (16 / 1) , 2 , (16 / 2), 4, (16 / 4)]
   *
   *  ******************* set is used to remove duplicate
   *  ******************* case 4 and (16 / 4) both equal to 4
   *  return factor set<int>.. this isn't ordered
   */

  for(var divisor = 1; divisor <= lowerBoundCheck; divisor++){
    if(N % divisor == 0){
      factors.add(divisor);
      factors.add(N ~/ divisor); // ~/ integer division 
    }
  }
  return factors;
}

答案 22 :(得分:0)

当我看到这个问题,即使numpy比python循环更快时,没人使用numpy时,我感到非常惊讶。通过使用numpy实现@agf的解决方案,结果平均快了 8倍。 我相信,如果您以numpy实施其他一些解决方案,那么您将获得美好的时光。

这是我的功能:

import numpy as np
def b(n):
    r = np.arange(1, int(n ** 0.5) + 1)
    x = r[np.mod(n, r) == 0]
    return set(np.concatenate((x, n / x), axis=None))   

请注意,x轴的数字不是功能的输入。功能输入为2,x轴上的数字减1。 因此,输入十是2 ** 10-1 = 1023

Performance test results of using numpy instead of for loops.

答案 23 :(得分:0)

我认为可读性和速度@ oxrock的解决方案是最好的,所以这里是为python 3 +重写的代码:

def num_factors(n):
    results = set()
    for i in range(1, int(n**0.5) + 1):
        if n % i == 0: results.update([i,int(n/i)])
    return results

答案 24 :(得分:-3)

我认为这是最简单的方法:

void show(const set<string>& s) {
    cout << "<";
    auto it = s.begin();
    if (it != s.end()) {
        cout << *it++;
    }
    for (; it != s.end(); ++it) {
        cout << "," << *it;
    }
    cout << ">" << endl;
}