CSP 試題編號201803-1 Java實現

import java.util.Scanner;

public class Main{

public static void main(String[] args) {
    Scanner input = new Scanner(System.in);
    int num = -1;
    int score = 0;
    int prior = 0;
    //當輸入0時  遊戲結束  計算分數
    while((num=input.nextInt())!=0){
        if(num==1){
            score = score+1;
            prior = 1;
        }
        else if(num==2){
            if(prior==0||prior==1){
                score = score+2;
                prior = 2;
            }
            else{
                score = score+prior+2;
                prior = prior+2;
            }
        }
    }
    System.out.println(score);
}

}