문제 설명
XYZ 마트는 일정한 금액을 지불하면 10일 동안 회원 자격을 부여합니다. XYZ 마트에서는 회원을 대상으로 매일 한 가지 제품을 할인하는 행사를 합니다. 할인하는 제품은 하루에 하나씩만 구매할 수 있습니다. 알뜰한 정현이는 자신이 원하는 제품과 수량이 할인하는 날짜와 10일 연속으로 일치할 경우에 맞춰서 회원가입을 하려 합니다.
예를 들어, 정현이가 원하는 제품이 바나나 3개, 사과 2개, 쌀 2개, 돼지고기 2개, 냄비 1개이며, XYZ 마트에서 15일간 회원을 대상으로 할인하는 제품이 날짜 순서대로 치킨, 사과, 사과, 바나나, 쌀, 사과, 돼지고기, 바나나, 돼지고기, 쌀, 냄비, 바나나, 사과, 바나나인 경우에 대해 알아봅시다. 첫째 날부터 열흘 간에는 냄비가 할인하지 않기 때문에 첫째 날에는 회원가입을 하지 않습니다. 둘째 날부터 열흘 간에는 바나나를 원하는 만큼 할인구매할 수 없기 때문에 둘째 날에도 회원가입을 하지 않습니다. 셋째 날, 넷째 날, 다섯째 날부터 각각 열흘은 원하는 제품과 수량이 일치하기 때문에 셋 중 하루에 회원가입을 하려 합니다.
정현이가 원하는 제품을 나타내는 문자열 배열want와 정현이가 원하는 제품의 수량을 나타내는 정수 배열number, XYZ 마트에서 할인하는 제품을 나타내는 문자열 배열discount가 주어졌을 때, 회원등록시 정현이가 원하는 제품을 모두 할인 받을 수 있는 회원등록 날짜의 총 일수를 return 하는 solution 함수를 완성하시오. 가능한 날이 없으면 0을 return 합니다.
내 코드 (python)
def solution(want, number, discount):
answer = 0
for i in range(len(discount)):
if discount[i] in want:
cnt=0
j=i
N=number[:]
while(1):
if discount[j] in want and N[want.index(discount[j])]!=0:
k=want.index(discount[j])
N[k]-=1
j+=1
cnt+=1
# print(i,": ",N," ",cnt)
if cnt==10:
answer+=1
break
if N[k]<0:
break
if j>=(len(discount)):
break
else:
break
else:
continue
return answer
for루프를 discount의 길이 만큼 돌리는데discount[i]가
want안에 있는 것과 일치 한다면 그에 해당하는 인덱스를 구하고number에서 그 인덱스에 해당하는 숫자를 -1 해준다.discount[i]를 시작으로 두고 그 다음에 해당하는 값을 나타내기 위해 j=i로 두고 j를 1씩늘린다.
while문이 돌아가는 과정에서 문제의 조건과 맞게 그 다음에 해당하는 것이want에 들어있지 않으면
while문을 종료하고 다음 값을 시작점으로 두고 다시for문을 돌린다. 10일 연속으로 일치할 경우르에 맞춰 회원가입을 한다고 했기 때문에 값이 일치할 때마다 올려줬던cnt값이 10이 되면answer+1 해준다.
끝으로answer를 return 해서 답을 낸다.
주석 처리 되어있는 print(i,": ",N," ",cnt)는 과정을 확인하기 위한 코드
c++코드는 추후에 추가 예정...!
컴퓨터공학 전공자가 아니라서 깔끔하고 효율적인 코드를 만들지 못했지만
꾸준히 하다보면 비슷하게라도 풀 수 있지 않을까
