class Solution {
public int solution(int n) {
int answer = 0;
if(n%2==1){
for(int i=1; i<=n; i+=2){
answer += i;
}
}else{
for(int i=2; i<=n; i+=2){
answer += i*i;
}
}
return answer;
}
}