15 lines
413 B
Python
15 lines
413 B
Python
|
def compute_roots_of_unity(primitive_root, p, n):
|
||
|
|
"""
|
||
|
|
Compute the roots of unity modulo p.
|
||
|
|
|
||
|
|
Parameters:
|
||
|
|
primitive_root (int): Primitive root modulo p.
|
||
|
|
p (int): Modulus.
|
||
|
|
n (int): Number of roots of unity to compute.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
list: List of roots of unity modulo p.
|
||
|
|
"""
|
||
|
|
roots_of_unity = [pow(primitive_root, i, p) for i in range(n)]
|
||
|
|
return roots_of_unity
|