「高手训练」手机游戏(monster)

前言

坑点还是蛮多的，不过既然自己做出来了就总结一下吧!

---

题目描述

明明的手机上有这样一个游戏，一排$n$个怪物，每个怪物的血量是$m_i$ 。
现在明明可以射出 $k$个伤害均为$p$的火球，当某个火球射到第$i$个怪物，除了这个怪物会掉血以外，它左边的第 j个怪物 ($j$ $\le$ $i$)，也会遭到$max(0,p-(i-j{)}^2)$的溅射伤害。
当某个怪物的血量为负的时候，它就死了，但它的尸体依然存在，即其他怪物不会因为它死而改变位置。
明明想用这 k个火球消灭掉所有的怪物，但他同时希望每个火球的伤害 能尽可能的小，这样他才能完美过关。
所有数均为整数。
输入格式
第一行两个数$n$,$k$ 。
第二行$n$个数 表示每个怪物的生命值$m_i$。
输出格式
一行一个整数表示最小的符合要求的$p$值。
样例
输入样例1
3 1
1 4 5
输出样例1
6
数据范围与提示
对于$30$%的数据,$n$ $\le$ $300$。
对于$100$%的数据,$n$ ,$k$ $\le$ $5e5$,$m_i$ $\le$ $1e10$
其中
内存限制：128 MiB
时间限制：1000 ms

网上的已有题解很多是$n^2\log \max m_i$假做法,因为一本通的OJ数据太水所以能过，但很明显这个时间复杂度是不对的。

显然火球大小对于能否通关来说是具有单调性的，所以考虑二分答案。
下界必定为1，上界的话考虑$k$最小，$m_i$最大，则只通过溅射伤害能够打到第一个怪物，需要10000000000+(499999)^2,即2e11左右。

但是这样做，$n\log n$的时间复杂度只有500000*40=2e7左右，似乎对于1s的时限有点小，那么肯定需要在$check函数$上做文章。

首先，显而易见的是我们每次需要从最后一个怪物开始打。

Key Point 1

其次，每次暴力枚举$p$是$n^2$的做法，那么我们考虑把每个数被操作后对左边的贡献是什么。

拆分一下题目中的$p$-$(i-j{)}^2$,即$p$-$i^2$-$j^2$+$2ij$,我们把$i^2$的和存下来，称作Sumi，这在后面做的贡献肯定是一定的。
把$i$的值存下来，称作Sum。
由乘法分配律,$2(i+k)j$=$2ij$+$2kj$,也就是说${\sum }_{i=j+1且做献}^{n}i\ast j$=Sum*j
$j^2$在遍历时处理即可。

Key Point 2

但是这样交上去明显不对，因为我们还没有计算被减去的贡献值，即，当$p-(i-j{)}^2$ $\le$ $0$时，它所做的贡献全部被消除，后面不再计算

Solution 1直接遍历时处理,但我们没办法存下每一处做贡献的i，这样做也是$n^2$的

还记得上面那句话吗

似乎对于1s的时限有点小。

我们惊喜的发现，贡献被抵消的位置也是具有单调性的，即如果$p-(i-j{)}^2$ $\le$ $0$，那么对于$k$ $<$ $j$ ,$p-(i-k{)}^2$必定小于 $0$。
那么我们可以再套一个二分，求出第一次贡献消除的位置。
Code

int search(int l,int r,int val,int op)//op指当前位置，val指外面一层二分的值
{
	while(l<r)
	{
		int mid=(l+r+1)>>1;
		if((op-mid)*(op-mid)>=val)
		{
			l=mid;//超过了，枚举更近的位置
		}
		else{
			r=mid-1;
		}
	}
	return l;
}

这样做，整体时间复杂度就是$nlogn\log \max m_i$$\approx$ $4e8$,但是数据是跑不满的，所以
开个c++11差不多是能过的

完整代码

#include <cstdio>
#include <iostream>
#include<cstring>
using namespace std;
#define int long long
int n, k, a[500005], r = 5e11, l = 1;
int Sumi, Sum, kk, sp,op,val[500005],S[500005],I[500005],cnt[500005];
bool pd;
int search(int l,int r,int val,int op)
{
	while(l<r)
	{
		int mid=(l+r+1)>>1;
		if((op-mid)*(op-mid)>=val)
		{
			l=mid;
		}
		else{
			r=mid-1;
		}
	}
	return l;
}
bool check(int now) {
    Sumi = 0, Sum = 0, kk = 0, sp = 0,op=0;//kk指用的火球个数，这样的好处是不会出现减去消除值后出现负数
    pd = 1;
    int X = 0;
    memset(val,0,sizeof(val));
    memset(S,0,sizeof(S));
    memset(I,0,sizeof(I));
    memset(cnt,0,sizeof(cnt));
    for (int i = n; i >= 1; i--) {
        X = a[i];
        sp-=val[i];//消除贡献
        Sumi-=S[i];
        Sum-=I[i];
        op+=cnt[i];
		X -=  sp - (Sumi +(kk-op) *i * i - 2 * Sum * i);//因为最多把前面所有贡献减完，所以最小为0
        if (X < 0) {
            continue;
        }
        int p = X / now;
        if (now * p <= X ) {
            p++;
        }
        if (kk + p > k) {
            pd = 0;
            break;
        }
        sp += p * now;//增加的p值
        kk += p;//使用数量
        Sumi += p * i * i;//存储i^2
        Sum += p * i;//存储i
        int kp=search(0,i-1,now,i);//第一次消除贡献的位置
        val[kp]+=p*now;//达到这个位置，贡献被消除
        S[kp]+=p * i * i;
        I[kp]+=p*i;
        cnt[kp]+=p;
    }
    return pd;
}
signed main() {
    scanf("%lld%lld", &n, &k);
    for (int i = 1; i <= n; i++) {
        scanf("%lld", &a[i]);
    }
    int ans=0;
    while (l <= r) {
        int mid = (l + r) >> 1;
        if (check(mid)) {
        	ans=mid;
            r = mid - 1;
        } else {
            l = mid + 1;
        }
    }
    printf("%lld", ans);
    return 0;
}

提交记录：

End