洛谷 P1485 火枪打怪

题目描述

LXL进入到了一片丛林,结果他发现有n只怪物排成一排站在他面前。LXL有一杆火枪能对付这些怪物。他知道从左至右数第i只怪物的血量是mi。现在LXL可以将一些子弹射向某个怪物。LXL可以控制他所发射的子弹数量及子弹的威力值。当某个子弹射到第i个怪物，如果这个子弹的威力值为p，除了这个怪物会掉p点血以外，它左边的第j个怪物(j<i)，也会遭到Max(0, p - (i - j) * (i - j))的溅射伤害（好神奇的子弹）。当某只怪物的血量小于0时，它就死了，但它的尸体还在，即怪物的位置永远不会改变。LXL希望只用k发子弹，请你求出一个最小的正整数p，使LXL用k发子弹且每发子弹的威力值为p就可以消灭所有怪物。

输入输出格式

输入格式：

第一行，两个正整数n,k。

第二行，n个正整数，第i个正整数表示从左至右数第i只怪物的血量mi。

输出格式：

一个正整数，表示子弹的最小威力值p。

输入输出样例

输入样例#1：

3 1
1 4 5

输出样例#1：

6

说明

对于30%的数据，n<=300。

对于100%的数据，n<=500000，k<=500000，1<=mi<=10^10

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

二分+数论~

神奇神奇又神奇~

数据范围很大，所以必须线性求出在枪射程是i的时候所需的子弹数，这里用到了公式推导，拆开后合并同类项，变成递推式，每次随着循环加上即可，注意射程范围之外的系数要减掉~

#include<cstdio>
#include<cstring>
#include<cmath>
#define ll long long

ll n,m,a[500001],c[500001],now[500001];

ll read()
{
	ll num=0,f=1;char s=getchar();
	while(s<'0' || s>'9') {if(s=='-') f=-1;s=getchar();}
	while(s>='0' && s<='9') {num=num*10+s-'0';s=getchar();}
	return num;
}

bool jud(ll u)
{
	ll lim=sqrt(u)+1,k1=0,k2=0,k3=0,tot,ans=0;
	memset(now,0,sizeof(now));
	for(int i=n;i;i--)
	{
		if(now[i+1])
		{
			k1+=now[i+1];
			k2+=now[i+1]*(i+1);
			k3+=now[i+1]*(i+1)*(i+1);
		}
		if(i+lim<=n && now[i+lim])
		{
			k1-=now[i+lim];
			k2-=now[i+lim]*(i+lim);
			k3-=now[i+lim]*(i+lim)*(i+lim);
		}
		tot=2*k2*i-k1*i*i-k3+u*k1;
		if(a[i]>=tot)
		{
			now[i]=(a[i]-tot)/u+1;
			ans+=now[i];
		}
		if(ans>m) return 0;
	}
	return 1;
}

int main()
{
	n=read();m=read();
	for(ll i=1;i<=n;i++) a[i]=read();
	ll l=1,r=1e14;
	while(l<r)
	{
		ll mid=(l+r)>>1;
		if(jud(mid)) r=mid;
		else l=mid+1;
	}
	printf("%lld\n",l);
	return 0;
}