문제

문자열 A와 B가 주어질 때, 문자열 B가 문자열 A에 포함되어 있는지를 조사하는 프로그램을 작성하시오. 단, 문자열 A와 B에는 알파벳으로만 이루어져 있으며, 공백은 포함되지 않는다고 가정한다.

---

입력

첫 번째 줄에 문자열 A, 두 번째줄에 문자열 B가 주어진다. 각각의 길이는 1,000을 넘지 않는다. 두 문자열은 모두 소문자 알파벳으로만 구성되어있다.

---

출력

문자열 B가 문자열 A에 포함되면 YES, 아니면 NO를 출력한다.

---

예제 입력

watermelon
melon

---

예제 출력

YES

---

아이디어

문자열 string을 입력받기 tmp와 비교하는 문제이다. if 문을 사용해 string[i]번째와 tmp[0]번째가 같을 경우 그 문자부터 len(tmp)까지의 문자를 비교하는 것이다. 예제 입력으로 예를 들어보면 string : watermelon , tmp : melon이다. 여기서 string[5]와 tmp[0]이 ‘m’으로 같다. 이렇게 같을 경우 string[5:10]이랑 tmp랑 비교하는 것이다. string[5:10] = ‘melon’이고 tmp도 ‘melon’이기 때문에 check를 ‘YES’로 바꾼 다음 마지막으로 check를 출력한다.

---

(수정) 더 짧은 코드를 구현했다. 간단하게 in 을 사용하여 작성했다. string을 입력받고 tmp를 입력 받고 in을 사용해서 tmp문자열이 string에 있으면 TRUE 없으면 FALSE 값을 넣는다.

코드

풀이보기(클릭)

string = input()
tmp = input()
check = 'NO'

for i in range(len(string)):
    if string[i] == tmp[0] and string[i:i+len(tmp)] == tmp:
        check = 'YES'

print(check)

---

풀이보기- 더 짧은 코드(클릭)

string = input()
tmp = input() in string
if tmp:
    print("YES")
else:
    print("NO")

---

문제 출처

ALGORITHM JOBS