5624. 좋은 수

1. 문제 링크
2. 참고 링크

1. 문제 해결


"질문 하기" 태그에 힌트가 주어져있다.
A+B+C = X, 즉 "i번째 수가 그 앞에 있는 수 세 개의 합" 을 수식으로 표현한 것이다.

A+B+C = X 는 A+B = X-C와 같은 의미이다.
즉, 현재 i번째 수를 기준으로 이전에 있는 수를 빼면서 나온 값(X-C)이 이전에 한번이라도 만든 적이 있다면 문제의 조건에 부합이 된다.

따라서, X를 값을 얻기 전에 A+B의 값을 채크한다.

2. 소스 코드

//좋은 수
import java.util.*;
import java.io.*;

public class Main {

	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		int [] array = new int[n];
		
		for(int i =0 ; i < n ; i ++) {
			array[i] = Integer.parseInt(st.nextToken());
		}
		
		int mid = 200000;
		boolean [] visited =new boolean[mid*2+1];
		
		int ans =0 ;
		for(int i =0 ; i < n ; i++) {
			for(int j =0 ; j < i ; j++) {
				if(visited[array[i] - array[j] + mid]) {
					ans++;
					break;
				}
			}
			
			for(int j =0 ; j <= i ; j++) {
				visited[array[i] + array[j] + mid ] = true;
			}
			
		}
		
		System.out.println(ans);
	}

}

좋은 웹페이지 즐겨찾기