2020牛客暑期多校训练营(第八场)K题Kabaleo Lite(单调栈,int_128)
2020牛客暑期多校训练营(第八场)K题Kabaleo Lite(单调栈,int_128)
题意: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("");
}
}