2020牛客暑期多校训练营(第八场)K题Kabaleo Lite(单调栈,int_128)

2020牛客暑期多校训练营(第八场)K题Kabaleo Lite(单调栈,int_128)

Kabaleo Lite

题意:n组数,有权值和个数,每次你可以从第一个数开始连续的取以获得其权值并且使个数减一(个数为0不可取),输出最大取的次数与其次数下最大的权值。

题解:最大次数为b[1]肯定没问题,然后最大值的策略如下,若个数满足的条件下前缀权值越大肯定取了最划算,所以对权值求前缀和,对个数来个前缀取min,然后用单调栈(权值递增)存入,然后直接算答案,可是答案会爆longlong,加个int_128就能过了。

#include<iostream>
#include<stack>
using namespace std;
#define ll long long
ll t,n,a[200007],b[200007];
stack<pair<ll,ll> >sa;
pair<ll,ll>lin;
inline void write(__int128 x)
{
    if(x<0)
    {
        putchar('-');
        x=-x;
    }
    if(x>9)
        write(x/10);
    putchar(x%10+'0');
}
int main(){
    ll o=0;
    scanf("%lld",&t);
    while(t--){
        o++;
        scanf("%lld",&n);
        a[0]=0;
        for(int i=1;i<=n;i++){
            scanf("%lld",&a[i]);
            a[i]+=a[i-1];
        }
        b[0]=1e9+7;
        for(int i=1;i<=n;i++){
            scanf("%lld",&b[i]);
            b[i]=min(b[i],b[i-1]);
        }
        lin={a[1],b[1]};
        sa.push(lin);
        for(int i=2;i<=n;i++){
            lin=sa.top();
            sa.pop();
            if(a[i]>lin.first){
                if(lin.second==b[i]){
                    sa.push({a[i],b[i]});
                }
                else{
                    lin.second-=b[i];
                    sa.push(lin);
                    sa.push({a[i],b[i]});
                }
            }
            else{
                sa.push(lin);
            }
        }
        __int128 ans=0,num1,num2;
        while(!sa.empty()){
            lin=sa.top();
            sa.pop();
            num1=lin.first;
            num2=lin.second;
            ans+=num1*num2;
        }
        printf("Case #%lld: %lld ",o,b[1]);
        write(ans);
        puts("");
    }
}

原文地址:https://www.cnblogs.com/whitelily/p/13430395.html