[프로그래머스] 숫자의 표현(Python, Java)

https://programmers.co.kr/learn/courses/30/lessons/12924

코딩테스트 연습 - 숫자의 표현

문제 설명

Finn은 요즘 수학공부에 빠져 있습니다. 수학 공부를 하던 Finn은 자연수 n을 연속한 자연수들로 표현 하는 방법이 여러개라는 사실을 알게 되었습니다. 예를들어 15는 다음과 같이 4가지로 표현 할 수 있습니다.

• 1 + 2 + 3 + 4 + 5 = 15
• 4 + 5 + 6 = 15
• 7 + 8 = 15
• 15 = 15

자연수 n이 매개변수로 주어질 때, 연속된 자연수들로 n을 표현하는 방법의 수를 return하는 solution를 완성해주세요.

제한사항

• n은 10,000 이하의 자연수 입니다.

입출력 예

n	result
15	4

입출력 예 설명

입출력 예#1
문제의 예시와 같습니다.

풀이 설명

1. 변수 s가 1부터 시작할때 ~ n부터 시작할때까지 반복한다.

  1-1. s가 n보다 작을때까지 s에 i를 더하고, i를 1씩 더한다.

  1-2. s가 n과 같다면 하나의 케이스가 존재하므로 answer에 1을 더한다.

* i = 1일때, s = 1 -> 3 -> 6 -> 10 -> 15 / i = 1 -> 2 -> 3 -> 4 -> 5

* i = 7일때, s = 7 -> 15 / i = 7 -> 8

Python3 코드

def solution(n):
    answer = 0
    for i in range(1, n+1):
        s = 0
        while s < n:
            s += i
            i += 1
        if s == n:
            answer += 1
    return answer

Java 코드

class Solution {
    public int solution(int n) {
        int answer = 0;
        int count;
        int a;
        for(int i=1; i<n+1; i++){
            count = 0;
            a = i;
            while(count < n){
                count += a;
                a++;
            }
            if(count == n)
                answer++;
        }
        return answer;
    }
}