27930번 : 당신은 운명을 믿나요? - Python

문제

https://www.acmicpc.net/problem/27930

풀이

KOREA 와 YONSEI를 저장한 배열을 0번째 부터 차레대로 비교하면서 수를 증가시키면 된다. 주변을 재거하고 뭐가 뭔저 나오느냐이기 때문에, 나올때마다 index를 증가시켜 먼저 index를 충족시키는 쪽이 정답이다.

코드

from sys import stdin

input = stdin.readline

korea = 'KOREA'
yonsei = 'YONSEI'

k = 0
y = 0

array = input()

for i in array:
    if i == korea[k] and i == yonsei[y]:
        k+=1
        y+=1
    elif i == korea[k]:
        k+=1
    elif i == yonsei[y]:
        y+=1

    if k==5:#KOREA의 길이
        print(korea)
        break
    elif y==6:#YONSEI의 길이
        print(yonsei)
        break