1부터 6까지 숫자가 적힌 주사위가 두 개 있습니다. 두 주사위를 굴렸을 때 나온 숫자를 각각 a, b라고 했을 때 얻는 점수는 다음과 같습니다.
- a와 b가 모두 홀수라면 a2 + b2 점을 얻습니다.
- a와 b 중 하나만 홀수라면 2 × (a + b) 점을 얻습니다.
- a와 b 모두 홀수가 아니라면 |a - b| 점을 얻습니다.
두 정수 a와 b가 매개변수로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.
- a와 b는 1 이상 6 이하의 정수입니다.
| a | b | result |
|---|---|---|
| 3 | 5 | 34 |
| 6 | 1 | 14 |
| 2 | 4 | 2 |
각 조건에 맞는 값 반환
class Solution {
public int solution(int a, int b) {
if (a % 2 != 0 && b % 2 != 0) return (a * a) + (b * b);
else if (a % 2 != 0 || b % 2 != 0) return (a + b) * 2;
else return Math.abs(a - b);
}
}접근법과 동일