1부터 6까지 숫자가 적힌 주사위가 두 개 있습니다.
두 주사위를 굴렸을 때 나온 숫자를 각각 a
, b
라고 했을 때 얻는 점수는 다음과 같습니다.
a
와 b
가 모두 홀수라면 a
² + b
² 점을 얻습니다.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 |
입출력 예 #1
입출력 예 #2
입출력 예 #3
class Solution {
public int solution(int a, int b) {
if (a % 2 == 1 && b % 2 == 1) {
return a * a + b * b;
} else if ((a % 2 == 1 && b % 2 != 1) || (a % 2 != 1 && b % 2 == 1)) {
return 2 * (a + b);
} else {
return Math.abs(a - b);
}
}
}