[BOJ] 7795번: 먹을 것인가 먹힐 것인가

https://www.acmicpc.net/problem/7795

 

문제

심해에는 두 종류의 생명체 A와 B가 존재한다. A는 B를 먹는다. A는 자기보다 크기가 작은 먹이만 먹을 수 있다. 예를 들어, A의 크기가 {8, 1, 7, 3, 1}이고, B의 크기가 {3, 6, 1}인 경우에 A가 B를 먹을 수 있는 쌍의 개수는 7가지가 있다. 8-3, 8-6, 8-1, 7-3, 7-6, 7-1, 3-1.

두 생명체 A와 B의 크기가 주어졌을 때, A의 크기가 B보다 큰 쌍이 몇 개나 있는지 구하는 프로그램을 작성하시오.

 

풀이

이중반복문을 사용할 경우, N과 M의 범위가 (1 ≤ N, M ≤ 20,000)이므로 최악의 경우 20000*2 만큼의 연산을 하기 때문에 시간초과가 일어날 것이다. 따라서 투포인터를 사용하여 풀었다

 

- 첫번째 시도

for _ in range(int(input())):
    cnt = 0

    N, M = map(int, input().split())
    A = list(map(int, input().split()))
    B = list(map(int, input().split()))

	#A와 B 정렬
    A.sort()
    B.sort()

	#A와 B 각각을 가리키는 포인터를 0으로 초기화 (투포인터)
    A_pointer, B_pointer = 0, 0
    
    while A_pointer < N:
        if A[A_pointer] > B[B_pointer]:
            cnt += 1

            if B_pointer + 1 < M:
                B_pointer += 1
            else:  #B_pointer == M - 1일 경우
                A_pointer += 1
                B_pointer = 0
            
        else: #A보다 B가 크거나 같은 경우
            A_pointer += 1
            B_pointer = 0

    print(cnt)

시간초과 발생! 조금 형태를 바꿈

 

for _ in range(int(input())):
    cnt = 0

    N, M = map(int, input().split())
    A = list(map(int, input().split()))
    B = list(map(int, input().split()))

    A.sort()
    B.sort()

    A_pointer, B_pointer = 0, 0
    
    while A_pointer < N:
        while B_pointer < M and A[A_pointer] > B[B_pointer]:
            cnt += 1
            B_pointer += 1
        A_pointer += 1
        B_pointer = 0

    print(cnt)

더 간결해진듯 보이지만 위 두개의 코드 모두 최악의 경우 20000*2의 연산을 가진다.

 

- 두번째 시도

for _ in range(int(input())):
    cnt = 0

    N, M = map(int, input().split())
    A = list(map(int, input().split()))
    B = list(map(int, input().split()))

    A.sort()
    B.sort()
    B_pointer = 0

    for A_pointer in range(N): #for문으로 A_pointer
        while B_pointer != M and A[A_pointer] > B[B_pointer]:
            B_pointer += 1
        cnt += B_pointer

    print(cnt)

내가보기엔 연산횟수가 고만고만한 거 같은데... 얘는 통과했다. 왜지!? 아는 분 있으면 댓글로 알려주세요^^~