CVBB ACM Team

import sympy.ntheory as nth
import random
 
 
if __name__ == '__main__':
    output = list()
    for mult in range(1, 120, 2):
        for exp in range(10, 63):
            num = mult << exp | 1
            if nth.isprime(num):
                output.append((num, mult, exp, nth.residue_ntheory.primitive_root(num)))
    output.sort()
    print(len(output))
    print('^         NTT 模数          ^   表达式   ^         原根          ^         NTT 模数          ^   表达式   ^         原根          ^')
    for i in range(0, len(output), 2):
        num, mult, exp, pr_root = output[i]
        string = f'|  ${num}$  |  ${mult}\\times{2}^{{{exp}}}+1$  |  ${pr_root}$  '
        if i + 1 < len(output):
            num, mult, exp, pr_root = output[i + 1]
            string += f'|  ${num}$  |  ${mult}\\times{2}^{{{exp}}}+1$  |  ${pr_root}$  '
        string += '|'
        print(string)