Submission #62829567

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
N,K = map(int,input().split())
A = list(map(int,input().split()))
# bigM = 1200000
bigM = 1000000 # N<10^6 だったので改めた
def solve():
    # 全てのA_iに対してカウントする配列を作る。計算量はO(N)
    arr1 = [0]*(bigM+1)
    for a_i in A:
        arr1[a_i] += 1
    # 全てのA_iの約数に対してカウントする配列を作る。計算量はO(N log N)
    arr2 = [0]*(bigM+1)
    arr2[1] = N  # 1 は全てのA_iの約数のため
    for p in range(2,bigM+1):
        for q in range(1,bigM+1):
            if p * q > bigM:
                break
            arr2[p] += arr1[p * q]  # pを約数とする数をarr2に取得
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

N,K = map(int,input().split())
A = list(map(int,input().split()))

# bigM = 1200000
bigM = 1000000 # N<10^6 だったので改めた

def solve():

    # 全てのA_iに対してカウントする配列を作る。計算量はO(N)
    arr1 = [0]*(bigM+1)
    for a_i in A:
        arr1[a_i] += 1

    # 全てのA_iの約数に対してカウントする配列を作る。計算量はO(N log N)
    arr2 = [0]*(bigM+1)
    arr2[1] = N  # 1 は全てのA_iの約数のため
    for p in range(2,bigM+1):
        for q in range(1,bigM+1):
            if p * q > bigM:
                break
            arr2[p] += arr1[p * q]  # pを約数とする数をarr2に取得

    # １～bigM までの組み合わせの数がarr2に入っているので、その数がK以上であれば使えるとしてarr3に約数を格納していく。計算量はO(N log N)
    arr3 = [1]*(bigM+1)  # この配列には約数が入るのだが、全ての整数は1を約数とするため、1を入れておく。
    for p in range(2,bigM+1):
        for q in range(1,bigM+1):
            if p * q > bigM:
                break
            if arr2[p] >= K:  # K回以上使用している場合
                arr3[p * q] = p  # p*qの約数としてpを代入。同じ数字(p*q)の約数としては後の値が優先される

    # 求まった約数を出力する
    for a in A:
        print(arr3[a])

solve()

Submission Info

Judge Result