백준/2018번/Java - 수들의 합

 

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

2018번: 수들의 합 5

---

시작인덱스와 종료인덱스 두 포인터 크기를 조정해가면서 수들의 합을 만족하는 경우를 찾는 문제이다.

연속된 자연수 중 가장 작은 수가 시작, 가장 큰 수가 종료이다.

맨 처음에는 시작, 종료 모두 1에서 시작하게 되므로 연속된 수의 합은 1이다.

합이 목표 숫자인 N보다 작으면 종료 포인터를 우측으로 옮겨(= 종료포인터의 크기가 +1된다.) 연속된 수의 합을 늘려주고

합이 목표 숫자인 N보다 크면 시작 포인터를 우측으로 옮겨(=시작포인터의 크기가 +1 된다.) 연속된 수의 합을 줄여준다.

import java.io.*;


public class Main {

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int N = Integer.parseInt(br.readLine());
        int start = 1, end = 1, sum = 1, count = 0;
        while(true) {
            if(end == N && start == end){
                count++;
                break;
            }
            if (sum > N) {
                sum -= start;
                start++;
            }
            else if (sum < N) {
                end++;
                sum += end;
            }
            else if (sum == N) {
                count++;
                end++;
                sum += end;
            }
        }
        bw.write(""+count);
        br.close();
        bw.flush();
        bw.close();
    }
}

반복문 조건을 end != N으로 놓고

count를 0이 아닌 1로 시작하는 방법도 있다.

import java.io.*;


public class Main {

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int N = Integer.parseInt(br.readLine());
        int start = 1, end = 1, sum = 1, count = 1;
        while(end != N) {
            if (sum > N) {
                sum -= start;
                start++;
            }
            else if (sum < N) {
                end++;
                sum += end;
            }
            else if (sum == N) {
                count++;
                end++;
                sum += end;
            }
        }
        bw.write(""+count);
        br.close();
        bw.flush();
        bw.close();
    }
}