[SWEA] 13038. 교환학생(python)

📌문제

💪아이디어

1. 수업시작일 정하기
   어떤 요일에 수업이 시작하느냐에 따라 머무르는 기간이 달라질 수 있다. 수업이 1인 날에 시작한다고 가정하고 총 머무르는 기간을 구하였다.
   
   –> 보완할점
   문자열재배치
   : 주에 수업이 시작하는 요일이 하나이고 이미 input_day[0]==’1’이거나 모든 요일에 수업을 하면 문자열 재배치를 하지 않아도 됨
   
   
2. 총 머무는 기간 계산하기
   일주일에 열리는 수업 수 만큼 일주일씩 머물어야한다. 이는 총 들어야하는 수업의 개수 // 일주일에 열리는 수업 수 로 몇 주 머무는지 계산할 수 있다.
   총 들어야 하는 수업의 개수 % 일주일에 열리는 수업 수만큼 며칠 더 머물건지 센다. 나머지 0이라면 마지막 수업이 해당 주 언제 끝나느냐에 따라 머무는 일수가 달라지기 때문에 몫에서 1을 빼준다. 들어야하는 수업 수 만큼 일주일에 수업열리는 리스트를 검사하면서 며칠 머무는 지 계산한다.
   
   

🥂코드

def cnt_minimum_stay(n,course_per_week,course_open_days,answer):
    week = n//course_open_days # 수업을 듣는 최대 주
    remaining_days=n%course_open_days
    if remaining_days==0: 
        week-=1
        remaining_days = course_open_days
    total= 7*week
    for chk in course_per_week:
        if chk=='1': remaining_days-=1
        total+=1
        if remaining_days==0: break

    if answer>total: answer=total
    return answer

T = int(input()) # 테스트 케이스 수 입력받기
for tc in range(1,T+1):
    n=int(input())
    input_days=input().split()
    answer=float('inf')
    course_open_days = input_days.count('1')#주 수업일 세기
    #학교에 최소로 머물러야하기 때문에 수업이 처음 시작하는 요일에 학교에 머물기 시작해야함.
    for i,chk in enumerate(input_days):
        if chk=='1': #수업 시작일 
            course_days = input_days[i:]+input_days[:i]
            answer = cnt_minimum_stay(n,course_days,course_open_days,answer)
    print('#{} {}'.format(tc,answer))