글 읽기 - python 4% 에서 틀렸습니다 뜹니다 ㅠ

girlera7 3년 전 0

반례 찾아주시면 정말 감사하겟씁니다.

'''
첫째 줄에는 세 개의 양의 정수 N, K, S가 빈칸을 사이에 두고 순서대로 주어진다. 첫 번째 정수 N은 아파트 단지의 수이며 2<=N<=30,000이다. 두 번째 정수 K는 1<=K<=2,000이며, 통학버스의 정원이다. 세 번째 정수 S는 학교의 위치를 나타낸다. 둘째 줄부터 N+1번째 줄에는 각 아파트 단지의 위치를 나타내는 정수와 이 단지에 사는 학생 수를 나타내는 정수가 빈칸을 사이에 두고 주어진다. 학교와 아파트 단지의 좌표는 0 이상 100,000 이하이며, 이 좌표들은 모두 서로 다르다. 각 아파트 단지의 학생 수는 1 이상 2,000 이하이다.
'''


class BusStop:

    def __init__(self, location, student_num):
        self.location = location
        self.student_num = student_num


def min_distance_for_visiting_all_bus_stop(bus_stops: [BusStop], school_location, bus_personnel):
    bus_stops.sort(key=lambda bus_stop: abs(bus_stop.location - school_location), reverse=True)

    min_distance = 0
    cur_person_at_bus = 0
    prev_location = school_location
    size = len(bus_stops)

    while size > 0:
        cur_bus_stop = bus_stops.pop(0)
        location = cur_bus_stop.location
        student_num = cur_bus_stop.student_num
        size -= 1

        min_distance += abs(location - prev_location)
        prev_location = location
        cur_buses = cur_person_at_bus + student_num
        remainder = cur_buses % bus_personnel
        share = cur_buses // bus_personnel

        if remainder == 0 and share >= 1:
            min_distance += (cur_buses // bus_personnel - 1) * 2 * abs(location - school_location)
            cur_buses = bus_personnel
        else:
            min_distance += (cur_buses // bus_personnel) * 2 * abs(location - school_location)
            cur_buses = remainder

        if cur_buses == bus_personnel:
            min_distance += (prev_location - school_location)
            prev_location = school_location
            cur_person_at_bus = 0
        else:
            cur_person_at_bus = remainder
            #print(min_distance, "min_distance")
            prev_location = location
       # print(prev_location, cur_person_at_bus)
        #print(min_distance)

    if cur_person_at_bus > 0:
        min_distance += abs(prev_location - school_location)

    return min_distance


apartmanet_num, bus_personnel, school_location = map(int, input().split())
# 학교보다 왼쪽에 있는 버스 정류장, 오른쪽에 있는 버스 정류장
# 더 적절한 이름이 생각이 안남.
left_bus_stops = []
right_bus_stops = []

for _ in range(apartmanet_num):
    location, student_num = map(int, input().split())
    if location < school_location:
        left_bus_stops.append(BusStop(location, student_num))
    else:
        right_bus_stops.append(BusStop(location, student_num))

print(min_distance_for_visiting_all_bus_stop(left_bus_stops, school_location, bus_personnel)
      + min_distance_for_visiting_all_bus_stop(right_bus_stops, school_location, bus_personnel))

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

python 4% 에서 틀렸습니다 뜹니다 ㅠ

girlera7 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말