글 읽기 - python 코드입니다!

dprjsgml 2년 전 0

어느부분이 틀렸을까요... ㅠㅠ

n,m,k=map(int,input().split())
shark=[]
arr=[[0 for _ in range(n)] for _ in range(n)]
for i in range(m):
    r,c,m,s,d=map(int,input().split())
    arr[r-1][c-1]=1
    shark.append([r-1,c-1,m,s,d])
dx=[-1,-1,0,1,1,1,0,-1]
dy=[0,1,1,1,0,-1,-1,-1]
sum=0
 
def fireball(r,c):
    global shark,arr
    len_shark=len(shark)
    shark1=[]
    m=0
    s=0
    cnt=0
    cnt_odd=0
    for i in range(len_shark):
        if shark[i][0]==r and shark[i][1]==c:
            m+=shark[i][2]
            s+=shark[i][3]
            cnt+=1
            if shark[i][4]%2==1: #홀수갯수세기 
                cnt_odd+=1
        else:
            shark1.append(shark[i])
    m//=5
    if m!=0 and s//cnt!=0:
        if cnt_odd==cnt or cnt_odd==0:
            shark1.append([r,c,m,s//cnt,0])
            shark1.append([r,c,m,s//cnt,2])
            shark1.append([r,c,m,s//cnt,4])
            shark1.append([r,c,m,s//cnt,6])
        else:
            shark1.append([r,c,m,s//cnt,1])
            shark1.append([r,c,m,s//cnt,3])
            shark1.append([r,c,m,s//cnt,5])
            shark1.append([r,c,m,s//cnt,7])
        arr[r][c]+=4
        arr[r][c]-=cnt
    shark=shark1

while k>0:
    len_shark=len(shark)
    for i in range(len_shark):#마법사상어 이동
        r,c,m,s,d=shark[i]
        arr[r][c]-=1
        if s>=n:
            s%=n
        nx=r+dx[d]*s
        ny=c+dy[d]*s
        if nx>=n:
            nx-=n
        if nx<0:
            nx+=n
        if ny>=n:
            ny-=n
        if ny<0:
            ny+=n
        shark[i][0]=nx
        shark[i][1]=ny
        arr[nx][ny]+=1
    for i in range(n):
        for j in range(n):
            if arr[i][j]>=2:
                fireball(i,j)
    k-=1

for i in range(len(shark)):
    sum+=shark[i][2]
print(sum)

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

문제

문제

출처

ICPC

python 코드입니다!

dprjsgml 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말