글 읽기 - 인덱스 에1러

lkyxbar 1년 전 0

11번째 줄에서 인덱스 에러가 발생하는데 왜 이러는 거죠?

import sys
from cmath import exp, pi
input=sys.stdin.readline 
def fft(a):
    L=len(a)
    if not L-1:
        return a
    even=fft(a[0::2])
    odd=fft(a[1::2])
    wN=[exp(2j*pi*i/L) for i in range(L>>1)]
    return [even[i]+wN[i]*odd[i] for i in range(L>>1)]+\
           [even[i]-wN[i]*odd[i] for i in range(L>>1)]

def ifft(a):
    L=len(a)
    if not L-1:
        return a
    even=ifft(a[0::2])
    odd=ifft(a[1::2])
    wN=[exp(-2j*pi*i/L) for i in range(L>>1)]
    return [even[i]+wN[i]*odd[i] for i in range(L>>1)]+\
           [even[i]-wN[i]*odd[i] for i in range(L>>1)]

def mul(vec1, vec2):
    L1, L2=len(vec1), len(vec2)
    n=1
    while n<L1+1 or n<L2+1:
        n+=n
    n+=n
    vec1+=[0]*(n-L1)
    vec2+=[0]*(n-L2)
    vec3=[0]*n
    vec1, vec2=fft(vec1), fft(vec2)
    for i in range(n):
        vec3[i]=vec1[i]*vec2[i]
    vec3=ifft(vec3)
    for i in range(n):
        vec3[i]=round(vec3[i].real/n)
    return vec3
N=int(input())
k=[int(input()) for _ in range(N)]
P=[0]*-~max(k)
P[0]=1
for i in k:
    P[i]=1
M=int(input())
d=[int(input()) for _ in range(M)]
ans=mul(P, P)
cnt=0
for i in d:
    cnt+=1 if ans[i] else 0
print(cnt)

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

인덱스 에1러

lkyxbar 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말