[프로그래머스]Lv.0 주사위 게임 2 C++ (set의 활용

https://school.programmers.co.kr/learn/courses/30/lessons/181930?language=cpp 

프로그래머스

소스코드1

#include <iostream>
#include <iomanip>
#include <cmath>
using namespace std;

int solution(int a, int b, int c) {
	int answer = 0;

	if (a == b && b == c)
		answer = (a + b + c) * (pow(a, 2) + pow(b, 2) + pow(c, 2)) * (pow(a, 3) + pow(b, 3) + pow(c, 3));
	else if (a == b || b == c || a == c)
		answer = (a + b + c) * (pow(a, 2) + pow(b, 2) + pow(c, 2));
	else 
		answer = (a + b + c);

	return answer;
}

소스코드2(set 활용)

#include <iostream>
#include <iomanip>
#include <cmath>
#include <set>
int solution(int a, int b, int c) {
	int answer = 0;

	answer = (a + b + c);
	std::set<int> s{a, b, c};
		 if (s.size() == 2)
	answer = (a + b + c) * (pow(a, 2) + pow(b, 2) + pow(c, 2));
			else if (s.size() == 1)
				answer = (a + b + c) * (pow(a, 2) + pow(b, 2) + pow(c, 2)) * (pow(a, 3) + pow(b, 3) + pow(c, 3));

	return answer;
}