神仙爱采药java代码

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        long n = sc.nextLong();//容量
        long a=0,b=0;//a是重量为1的药的个数，b是重量为2个药的个数
        long v = 0;//背包当前重量
        long sum = 0;//总药丸数

        String s = sc.next();//每种药的安排

        for(int i = 0;i<s.length();++i){//遍历每一天
            if(s.charAt(i)=='1'){//当前为1格药
                if(n-v >=1){//背包容量多于1个格子
                    a++;//1格药直接放进来
                    v++;//背包重量+1
                }else{//如果背包容量不足
                    if(b>0){//而且存在2格药
                        b--;//2格药的数量-1
                        v-=2;//移出一个2格药
                        a++;//放入1格药
                        v++;//背包重量+1
                    }
                }
            }else if(s.charAt(i)=='2'){//当前为2格药
                if(n-v>=2){//如果背包容量大于2
                    b++;//就把2格药直接放进来
                    v+=2;//再把背包重量+2
                }
            }
            long num = (a+b);//完成当天的采集后，两种药的数量相加
            sum+=num;//然后累加
        }
        System.out.println(sum);
    }
}