[백준 1292번] 쉽게 푸는 문제

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

1292번: 쉽게 푸는 문제

 

문제

동호는 내년에 초등학교를 입학한다. 그래서 동호 어머니는 수학 선행 학습을 위해 쉽게 푸는 문제를 동호에게 주었다.

이 문제는 다음과 같다. 1을 한 번, 2를 두 번, 3을 세 번, 이런 식으로 1 2 2 3 3 3 4 4 4 4 5 .. 이러한 수열을 만들고 어느 일정한 구간을 주면 그 구간의 합을 구하는 것이다.

하지만 동호는 현재 더 어려운 문제를 푸느라 바쁘기에 우리가 동호를 도와주자.

입력

첫째 줄에 구간의 시작과 끝을 나타내는 정수 A, B(1 ≤ A ≤ B ≤ 1,000)가 주어진다. 즉, 수열에서 A번째 숫자부터 B번째 숫자까지 합을 구하면 된다.

출력

첫 줄에 구간에 속하는 숫자의 합을 출력한다.

예제 입력 1 복사

3 7

예제 출력 1 복사

15

 

문제 분석

1. 숫자n이 연달아 n번 나오는 수열을 구현합니다. 

    인덱스 0은 필요 없습니다.

    크기 1000까지 숫자를 다 넣습니다. 

2. 입력받은 수 : a, b

3.  a ~ b에 있는 숫자를 더해서 출력(sum)합니다.

 

 

코드 구현 핵심

💡 숫자 n이 연달아 n번 나오는 수열을 구현합니다. 
    이중 for문 활용

 

실패 코드 

import java.util.*;

public class Main {
	public static void main(String[] args) {
            int[] arr = new int[1001];
            for(int i=1; i<arr.length; i++) { 
            	for(int j=1; j<=i; j++) { 
                 arr[i] = i; 
                }
            }
    		System.out.println(Arrays.toString(arr));
	}
 
}

결과

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, .... 1000]

 

Array로 구현하려고 하니까 고정 인덱스에 n만큼 n번 넣는 게 힘들었습니다.

 

 

이번에는 ArrayList를 통해 list.add로 구현해보겠습니다.

 

 

성공 코드

import java.util.*;

public class Main {
	public static void main(String[] args) {
            Scanner kb = new Scanner(System.in);
            int a = kb.nextInt();
            int b = kb.nextInt();
            int sum = 0;
            // 수열을 담을 배열
            List<Integer> list = new ArrayList<>();
            // for문을 돌면서 1000까지만 받는 것을 확인 할 변수
            int cnt = 0;
            
            // [[ 숫자 n이 연달아 n번 나오는 수열 구현 ]]
            for (int i = 1; i < 1001; i++) {
                for (int j = 0; j < i; j++) {
                    if(cnt == 1000) break;
                    list.add(i);
                    cnt++;
                }
            }
   
           for(int i=a-1; i<=b-1; i++) {
               sum += list.get(i) ;
           }
           System.out.println(sum);
	}
 
}

 

---

 

ArrayList 전체 출력하는 방법

 // for문을 통한 전체출력
    for(Integer i : list) { 
        System.out.print(i + " ");
    }
    System.out.println("cnt : " + cnt);

 

결과

1 2 2 3 3 3 4 4 4 4 5 5 5 5 5 6 6 6 6 6 6 7 7 7 7 7 7 7 .... 45 cnt : 1000