[c++] 프로그래머스 - 약수의 개수와 덧셈(Level 1)

문제

코딩테스트 연습 - 약수의 개수와 덧셈

 

코드

#include <string>
#include <vector>
#include <map>


using namespace std;

int solution(int left, int right) {
    int answer = 0;
    map<int,int>divisor;
    for(int i=1; i<1000; i++){	// 약수의 갯수 찾기
        for(int k=1; k<=i; k++){
            if(i%k==0) divisor[i]++;
        }
    }
    
    for(int i=left; i<=right; i++){	// 약수의 갯수가 홀수면 빼고, 짝수면 더한다.
        if(divisor[i]&1) answer-=i;
        else answer+=i;
    }
    return answer;
}

 

풀이

새벽 5시 21분. 잠이 오지 않아서 오랜만에 프로그래머스에 접속하여 간단한 문제 하나를 찾아서 풀었습니다. 최대 입력값이 1000까지로 작은 수가 들어오기 때문에 O(n^2)으로 먼저 각 수마다 약수의 개수를 전부 찾아서 저장한 뒤 홀수면 빼고, 짝수면 더하는 식으로 풀이하였습니다.