[백준 1484] 다이어트 (C++)

문제 링크: www.acmicpc.net/problem/1484

1484번: 다이어트

1. 주의할 점

- a*a - b*b = (a+b)*(a-b)로 접근한다

 

2. 구현

- G = (a+b)*(a-b)라고 생각하여 G의 약수중에서 작은 값을 V벡터에 넣는다. 즉, a-b의 값을 V에 너흔ㄴ다

- V에 있는 값을 통해 a+b를 구한다.

- {(a+b) + (a-b)}/2 = a 식을 통해 a를 구한다. 이때, 좌변은 2로 나누기 전, 짝수여야 한다

- 모든 a 값을 ans벡터에 넣고 정렬한 후, 출력한다

 

#include <iostream>
#include <vector>
#include <math.h>
#include <algorithm>
using namespace std;

int main() {
    int num,result=0,small,big;
    cin>>num;
    vector<int> v,ans;
    for(int i=1;i<=sqrt(num);i++){
        if(num%i==0){
            if(num/i == i) continue;
            v.push_back(i);
        }
    }
  
    int len=v.size();
    for(int i=0;i<len;i++){
        int small = v[i];
        int big = num/small;
        if((big+small)%2==1) continue;
        ans.push_back((big+small)/2);
    }
    sort(ans.begin(),ans.end());
    if(ans.empty()) cout<<-1;
    else{
        for(int i=0;i<ans.size();i++)
            cout << ans[i]<<'\n';
    }   
    return 0;
}