반응형
투 포인트 문제 풀기!
문제
n개의 서로 다른 양의 정수 a1, a2,..., an으로 이루어진 수열이 있다. ai의 값은 1보다 크거나 같고, 1000000보다 작거나 같은 자연수이다. 자연수 x가 주어졌을 때, ai + aj = x (1 ≤ i < j ≤ n)을 만족하는 (ai, aj) 쌍의 수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 수열의 크기 n이 주어진다. 다음 줄에는 수열에 포함되는 수가 주어진다. 셋째 줄에는 x가 주어진다. (1 ≤ n ≤ 100000, 1 ≤ x ≤ 2000000)
출력
문제의 조건을 만족하는 쌍의 개수를 출력한다.
예제 입력 1
9 5 12 7 10 9 1 2 3 11 13
예제 출력 1
3
코드 제출
n = int(input()) # 원소의 개수
numbers = sorted(list(map(int, input().split()))) # 원소 리스트
sum = int(input()) # 목표 수
start, end = 0, n-1
count = 0
while start < end:
temp = numbers[start] + numbers[end]
if temp == sum:
count += 1
start += 1
elif temp > sum:
end -= 1
else:
start += 1
print(count)
* 양 쪽을 기준으로 투 포인트를 쓰려면 정렬이 필요하다. 이 문제에서는 합을 만족하는 수의 개수만 세면 되기 때문에
배열을 정렬한 후 문제를 풀면 된다. 하지만 만약 배열의 인덱스를 묻는 문제라면, 정렬을 할 경우 순서가 뒤죽박죽 되니 주의하자!.
투 포인트를 둘 다 앞에서부터 시작하려다가 사실 한 번 틀렸다.. , 그냥 정렬을 한 후 양 끝에서 하면 편하다.
반응형
'algorithm' 카테고리의 다른 글
[BAEKJOON] 1644번: 소수의 연속합 - 파이썬(python) (0) | 2021.07.04 |
---|---|
[BAEKJOON] 1806번: 부분합 - 파이썬(python) (0) | 2021.07.01 |
[BAEKJOON] 4949번: 균형잡힌 세상 - 파이썬(python) (0) | 2021.06.29 |
[프로그래머스 level1] 로또의 최고 순위와 최저 순위 - 파이썬 (0) | 2021.06.20 |
[프로그래머스 level1] 실패율 - 파이썬 (0) | 2021.06.19 |