Competition/Baekjoon
[백준] 7453번 자바 합이 0인 네 정수
bisi
2020. 4. 30. 20:28
문제 출처
https://www.acmicpc.net/problem/7453
7453번: 합이 0인 네 정수
문제 정수로 이루어진 크기가 같은 배열 A, B, C, D가 있다. A[a], B[b], C[c], D[d]의 합이 0인 (a, b, c, d) 쌍의 개수를 구하는 프로그램을 작성하시오. 입력 첫째 줄에 배열의 크기 n (1 ≤ n ≤ 4000)이 주어진다. 다음 n개 줄에는 A, B, C, D에 포함되는 정수가 공백으로 구분되어져서 주어진다. 배열에 들어있는 정수의 절댓값은 최대 228이다. 출력 합이 0이 되는 쌍의 개수를 출력한다. 예제 입력 1 복
www.acmicpc.net
접근 방식 및 풀이
- 1208번과 비슷한 문제
[Algorithm/백준] - [백준] 1208번 자바 부분수열의 합2
- 4개의 배열을 2개의 배열의 부분합으로 만든 후, 2개의 배열을 투포인트 알고리즘을 적용한다.
소스 코드
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
import java.util.Arrays;
public class Main {
static long[][] arr;
static long[] arrab, arrcd;
static long count=0;
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
arr = new long[n][4];
arrab = new long[n*n];
arrcd = new long[n*n];
for (int i = 0; i <n ; i++) {
arr[i][0] = sc.nextLong();
arr[i][1] = sc.nextLong();
arr[i][2] = sc.nextLong();
arr[i][3] = sc.nextLong();
}
// 두배열의 부분합 구하기
int k =0;
for (int i = 0; i <n ; i++) {
for (int j = 0; j <n ; j++) {
arrab[k] = arr[i][0] + arr[j][1];
arrcd[k] = arr[i][2] + arr[j][3];
k++;
}
}
int leftIdx = 0;
int rightIdx = (n*n) -1;
while (leftIdx < (n*n) && rightIdx>=0){// 전체 범위 제한둠
long lv = arrab[ leftIdx];
long rv = arrcd[rightIdx];
if(lv+rv ==0){
long lc = 0;
while(leftIdx < arrab.length && arrab[leftIdx] == lv){
lc++;
leftIdx++;
}
long rc = 0;
while(rightIdx >= 0 && arrcd[rightIdx] == rv){
rc++;
rightIdx--;
}
count += (rc*lc);
}
if(lv+rv>0){
rightIdx--;
}
if(lv+rv<0){
leftIdx++;
}
}
// 4개의 배열을 모두 사용하여 만들었기 때문에 공집합인 경우는 없다.
System.out.println(count);
}
}
|
결과
