选址 [差分]

最新推荐文章于 2022-07-08 18:41:30 发布

Zbr162

最新推荐文章于 2022-07-08 18:41:30 发布

阅读量163

点赞数

CC 4.0 BY-SA版权

分类专栏：基础算法-差分 Second

本文链接：https://blog.csdn.net/Zbr162/article/details/98057033

Second 同时被 2 个专栏收录

109 篇文章

订阅专栏

基础算法-差分

2 篇文章

订阅专栏

$选址$

$正解部分\color{red}{正解部分}$

考虑什么时候 $i$ 会对 $x$ 产生贡献,
当 $c_i-|x-i|^2 > 0$ 时, 会产生贡献, 解方程得到 $x∈(i−ci,i+ci)x∈(i-\sqrt{c_i}, i+\sqrt{c_i})$ .

$∴\therefore$ 当 $x∈(i−ci,i+ci)x∈(i-\sqrt{c_i}, i+\sqrt{c_i})$ 时, $i$ 会对 $x$ 产生贡献 .

贡献为 $c_i-i^2 + 2ix - x^2$ ,

$c_i-i^2$ 为常数, 可以直接使用一个差分数组处理 .
$2 i * x$ 的系数 $2 i$ 使用一个差分数组处理 .
$1*x^2$ 的系数 $1$ 使用一个差分数组处理 .

$实现部分\color{red}{实现部分}$

#include<bits/stdc++.h>
#define reg register
typedef long long ll;

const int maxn = 100005;

int N;

ll C[maxn];
ll cf1[maxn];
ll cf2[maxn];
ll cf3[maxn];

int main(){
        scanf("%d", &N);
        for(reg int i = 1; i <= N; i ++) scanf("%lld", &C[i]);
        for(reg int i = 1; i <= N; i ++){
                int l = std::max(1, (int)(i*1.0-sqrt(C[i])+1.0));
                int r = std::min(N, (int)(i*1.0+sqrt(C[i])));
                cf1[l] += C[i] - 1ll*i*i,
                cf1[r+1] -= C[i] - 1ll*i*i;
                cf2[l] += 2ll*i, cf2[r+1] -= 2ll*i;
                cf3[l] ++, cf3[r+1] --;
        }
        ll a = 0, b = 0, c = 0;
        for(reg int i = 1; i <= N; i ++){
                a += cf1[i], b += cf2[i], c += cf3[i];
                printf("%lld ", a + b*i - c*i*i);
        }
        return 0;
}