bzoj5108 数据_BZOJ1835: [ZJOI2010]base 基站选址【线段树优化DP】

Description

有N个村庄坐落在一条直线上，第i(i>1)个村庄距离第1个村庄的距离为Di。需要在这些村庄中建立不超过K个通讯基站，在第i个村庄建立基站的费用为Ci。如果在距离第i个村庄不超过Si的范围内建立了一个通讯基站，那么就成它被覆盖了。如果第i个村庄没有被覆盖，则需要向他们补偿，费用为Wi。现在的问题是，选择基站的位置，使得总费用最小。 输入数据 (base.in) 输入文件的第一行包含两个整数N,K，含义如上所述。 第二行包含N-1个整数，分别表示D2,D3,…,DN ，这N-1个数是递增的。 第三行包含N个整数，表示C1,C2,…CN。 第四行包含N个整数，表示S1,S2,…,SN。 第五行包含N个整数，表示W1,W2,…,WN。

Input

输出文件中仅包含一个整数，表示最小的总费用。

Output

3 2 1 2 2 3 2 1 1 0 10 20 30

Sample Input

4

Sample Output

40%的数据中，N<=500；

100%的数据中，K<=N，K<=100，N<=20,000，Di<=1000000000，Ci<=10000，Si<=1000000000，Wi<=10000

思路

首先一个思路是前i个节点放j个基站

为了方便转移，强制第i个节点上一定会放上一个基站

然后怎么办？

有一个转移是\(dp_{i,k}=dp_{j,k}+calc(j+1,i-1)+c_i\)

然后考虑怎么快速计算对于每个i的\(dp_{j,k}+calc(j+1,i-1)\)

我们令每一个节点可以覆盖到它的基站位置是\([pre_i,nxt_i]\)

那么\(calc(j+1,i-1)=\sum_{p=j+1}^{i-1}[pre_i>j\&\&nxt_i

如果把i向右平移，发现受影响的新增节点只有\(nxt_t=i\)的所有节点

那么就按照\(nxt_t\)排序，每次因为我们维护所有的\(dp_{j,k}+calc(j+1,i-1)\)

所以对于所有\([1,pre_t-1]\)的dp值全部要加上\(w_t\)

然后就用线段树维护一下就可以了

注意当k是0的时候初值的处理

然后应该没什么大问题了

是我傻逼写错了线段树。。

//Author: dream_maker

#include

using namespace std;

//----------------------------------------------

typedef pair pi;

typedef long long ll;

typedef double db;

#define fi first

#define se second

#define fu(a, b, c) for (int a = b; a <= c; ++a)

#define fd(a, b, c) for (int a = b; a >= c; --a)

#define fv(a, b) for (int a = 0; a < (signed)b.size(); ++a)

const int INF_of_int = 1e9;

const ll INF_of_ll = 1e18;

template

void Read(T &x) {

bool w = 1;x = 0;

char c = getchar();

while (!isdigit(c) && c != '-') c = getchar();

if (c == '-') w = 0, c = getchar();

while (isdigit(c)) {

x = (x<<1) + (x<<3) + c -'0';

c = getchar();

}

if (!w) x = -x;

}

template

void Write(T x) {

if (x < 0) {

putchar('-');

x = -x;

}

if (x > 9) Write(x / 10);

putchar(x % 10 + '0');

}

//----------------------------------------------

const int N = 1e5 + 10;

int n, k, c[N], d[N], s[N], w[N];

int dp[2][N];

struct Node {

int pre, nxt, id;

} p[N];

bool operator < (const Node a, const Node b) {

return a.nxt < b.nxt;

}

#define LD (t << 1)

#define RD (t << 1 | 1)

int val[N << 2], tag[N << 2];

void pushnow(int t, int vl) {

val[t] += vl;

tag[t] += vl;

}

void pushup(int t) {

val[t] = min(val[LD], val[RD]);

}

void pushdown(int t) {

if (tag[t]) {

pushnow(LD, tag[t]);

pushnow(RD, tag[t]);

tag[t] = 0;

}

}

void build(int t, int l, int r) {

tag[t] = 0;

val[t] = INF_of_int;

if (l == r) return;

int mid = (l + r) >> 1;

build(LD, l, mid);

build(RD, mid + 1, r);

}

void modify(int t, int l, int r, int pos, int vl) {

if (l == r) {

val[t] = vl;

return;

}

pushdown(t);

int mid = (l + r) >> 1;

if (pos <= mid) modify(LD, l, mid, pos, vl);

else modify(RD, mid + 1, r, pos, vl);

pushup(t);

}

void modify(int t, int l, int r, int ql, int qr, int vl) {

if (ql > qr) return;

if (ql <= l && r <= qr) {

pushnow(t, vl);

return;

}

int mid = (l + r) >> 1;

pushdown(t);

if (qr <= mid) modify(LD, l, mid, ql, qr, vl);

else if (ql > mid) modify(RD, mid + 1, r, ql, qr, vl);

else {

modify(LD, l, mid, ql, mid, vl);

modify(RD, mid + 1, r, mid + 1, qr, vl);

}

pushup(t);

}

int find_pre(int x) {

int l = 1, r = x, res = 0;

while (l <= r) {

int mid = (l + r) >> 1;

if (s[x] >= d[x] - d[mid]) res = mid, r = mid - 1;

else l = mid + 1;

}

return res;

}

int find_nxt(int x) {

int l = x, r = n, res = 0;

while (l <= r) {

int mid = (l + r) >> 1;

if (s[x] >= d[mid] - d[x]) res = mid, l = mid + 1;

else r = mid - 1;

}

return res;

}

int main() {

#ifdef dream_maker

freopen("input.txt", "r", stdin);

freopen("output.txt", "w", stdout);

#endif

memset(dp, 0, sizeof(dp));

Read(n), Read(k);

fu(i, 2, n) Read(d[i]);

fu(i, 1, n) Read(c[i]);

fu(i, 1, n) Read(s[i]);

fu(i, 1, n) Read(w[i]);

fu(i, 1, n) {

p[i].pre = find_pre(i);

p[i].nxt = find_nxt(i);

p[i].id = i;

}

sort(p + 1, p + n + 1);

int ans = INF_of_int, now = 1, sumw = 0;

int ind = 0;

fu(i, 1, n) {

dp[ind][i] = sumw + c[i];

while (now <= n && p[now].nxt <= i)

sumw += w[p[now].id], ++now;

}

fu(j, 2, k + 1) {

build(1, 1, n);

ind ^= 1, now = 1;

fu(i, 1, n) {

dp[ind][i] = val[1] + c[i];

while (now <= n && p[now].nxt <= i)

modify(1, 1, n, 1, p[now].pre - 1, w[p[now].id]), ++now;

modify(1, 1, n, i, dp[ind ^ 1][i]);

}

ans = min(ans, val[1]);

}

Write(ans);

return 0;

}