import random import sys # 표준 출력을 파일로 리다이렉트하여 테스트 케이스를 파일에 저장 # 사용법: python generator.py > input.txt # 또는 그냥 콘솔에 출력하려면 이 두 줄을 주석 처리 # sys.stdout = open("input.txt", "w") # --- 설정값 (여기서 원하는대로 변경) --- # L: 도로의 최대 좌표 (0부터 L까지) # 문제 조건: 1 <= L <= 10**18 # 디버깅을 위해 작은 값부터 시작하여 점차 키워보는 것을 추천 L_val = 1000000000000000000 # L의 최대값 (10^18) # L_val = 10 # 작은 L로 테스트 # N: 가로등 개수 # 문제 조건: 1 <= N <= 200000 N_val = 200000 # N의 최대값 # N_val = 2 # 작은 N으로 테스트 # K: 출력할 어두운 정도의 개수 # 문제 조건: 1 <= K <= L+1 (K는 L+1을 넘지 않음) # K <= 200000 (이건 N과 K의 제약이 있을 때) # K_val = L_val + 1 # K의 최대값 K_val = 100000 # K의 최대값 (K는 N보다 클 수도, 작을 수도 있음) # K_val = 5 # 작은 K로 테스트 # 가로등 위치 생성 방식 # 1: 무작위 위치 # 2: 균등 간격 위치 # 3: 특정 구간에 몰린 위치 # 4: 양 끝에 몰린 위치 LIGHTS_GEN_TYPE = 1 # --- 제너레이터 코드 시작 --- # L, N, K 출력 print(L_val, N_val, K_val) # 가로등 위치 생성 lights = set() # 중복 방지를 위해 set 사용 if LIGHTS_GEN_TYPE == 1: # 무작위 위치 while len(lights) < N_val: lights.add(random.randint(0, L_val)) # 0부터 L까지 무작위 위치 elif LIGHTS_GEN_TYPE == 2: # 균등 간격 위치 if N_val > 1: step = L_val // (N_val - 1) for i in range(N_val): lights.add(i * step) else: lights.add(random.randint(0, L_val)) # N=1일 경우 무작위 elif LIGHTS_GEN_TYPE == 3: # 특정 구간에 몰린 위치 (예: 중앙에 몰림) if N_val > 0: start_range = max(0, L_val // 2 - N_val * 5) end_range = min(L_val, L_val // 2 + N_val * 5) while len(lights) < N_val: lights.add(random.randint(start_range, end_range)) else: lights.add(random.randint(0, L_val)) # N=1일 경우 무작위 elif LIGHTS_GEN_TYPE == 4: # 양 끝에 몰린 위치 if N_val > 0: for i in range(N_val // 2): lights.add(i) # 0부터 N_val/2-1까지 for i in range(N_val // 2, N_val): lights.add(L_val - (N_val - 1 - i)) # L_val부터 L_val - (N_val/2 -1)까지 while len(lights) < N_val: # 혹시 모를 중복으로 N_val 안 채워질 경우 lights.add(random.randint(0, L_val)) else: lights.add(random.randint(0, L_val)) # N=1일 경우 무작위 # 가로등 위치 정렬 및 출력 sorted_lights = sorted(list(lights)) print(*sorted_lights) # --- 제너레이터 코드 끝 ---

코드 #6582