글 읽기 - 혹시 코드좀 봐주실 수 있을까요?

blackberry97 2년 전 0

제가 생각한건 이렇습니다.

우선 anta 와 tica는 반드시 문자열에 포함 되어야 하니깐

a,n,t,i,c 5개 알파벳은 반드시 배워야 합니다.

따라서 k가 5보다 작으면 위 조건을 만족하지 못하기 때문에 0을 출력하고

k가 5 이상일 경우 입력받은 문자열에서 anta 와 tica를 앞뒤에서 제거하고 각 문자열 안에서 중복되는 문자열을 제거 합니다.

그럼 남은 문자열의 개수가 적은 순서대로 만약에 arr배열 안에 해당 알파벳이 있을경우 arr에 알파벳을 추가하고 만약 k의 길이보다 커지면 break 를 해서 cnt를 출력합니다.

arr안에 존재할경우 c를 1씩 증가시켜주고 만약 c가 입력받은 문자열의 길이와 같아지면 cnt를 1씩 증가시켜주는 방식으로 나갑니다.

반례도 다 통과하는것 같은데 1%에서 틀렸습니다라고 나오네요ㅠㅠ

import sys

n,k = map(int,sys.stdin.readline().split())

arr = ['a', 'n', 't', 'i', 'c']
ans = []
for i in range(n):
    tmp = list(str(sys.stdin.readline().rstrip()))
    ans.append(list(set(tmp[4:len(tmp) - 4])))

if k < 5:
    print(0)

elif k >= 5:
    for i in range(len(ans)):
        x = ans[i].copy()
        for j in range(len(ans[i])):
            if ans[i][j] == 'a' or ans[i][j] == 'n' or ans[i][j] == 't' or ans[i][j] == 'i' or ans[i][j] == 'c':
                x.remove(ans[i][j])
        ans[i] = x

    ans.sort(key=lambda x:(len(x),x))

    cnt = 0
    check = 0

    for i in ans:
        c = 0
        i.sort()
        for j in i:
            if j not in arr:
                arr.append(j)
                if len(arr) > k:
                    arr.pop()
                    check = 1
                    break
                else:
                    c = c + 1
                    check = 0

            elif j in arr:
                c = c + 1
                pass
        if check == 1:
            break
        elif check == 0 and c == len(i):
            cnt = cnt + 1

    print(cnt)

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

문제

문제

출처

ICPC

혹시 코드좀 봐주실 수 있을까요?

blackberry97 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말