难点就在于check的实现,wa了很多发以后,选择了分情况讨论,核心就是一个递推的思想,理清楚情况就行。
代码如下
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
long long a[maxn];
int n;
long long m;
long long c[maxn];
bool check(long long x)
{
memset(c,0,sizeof(long long)*(n+2));
long long all=m-1;
c[1]=a[1];
for(int i=1;i<=n;i++)
{
if(i==n)
{
if(x<=c[i])
return true;
long long now=(x-c[i]+a[i]-1)/a[i];
if(all>=now*2-1)
return true;
return false;
}
else if(i==n-1)
{
if(x<=c[i])
continue;
long long now=(x-c[i]+a[i]-1)/a[i];
if(all>=now*2)
{
all-=now*2;
c[i]+=now*a[i];
c[i+1]+=now*a[i+1];
}
else return false;
}
else
{
if(x<=c[i])
{
if(all)
{
c[i+1]+=a[i+1];
all--;
continue;
}
else return false;
}
long long now=(x-c[i]+a[i]-1)/a[i];
if(all>=now*2+1)
{
all-=(now*2+1);
c[i]+=now*a[i];
c[i+1]+=(now+1)*a[i+1];
}
else return false;
}
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%lld",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
}
if(m==0)
{
printf("0\n");
continue;
}
long long l=0,r=1e17;
long long ans=0;
while(l<r)
{
long long mid=(l+r)/2;
if(check(mid))
{
l=mid+1;
ans=mid;
}
else r=mid;
}
printf("%lld\n",ans);
}
return 0;
}