APIO2010 特别行动队

APIO2010 特别行动队

【题目描述】

   你有一支由 n 名预备役士兵组成的部队，士兵从1到n编号，要将他们拆分成若干特别行动队调入战场。出于默契的考虑，同一支特别行动队中队员的编号应该连续，即为形如(i, i + 1, …, i + k)的序列。

    编号为 i 的士兵的初始战斗力为 xi ，一支特别行动队的初始战斗力 x为队内士兵初始战斗力之和，即 x = xi + xi+1 + … + xi+k。
　　通过长期的观察，你总结出一支特别行动队的初始战斗力 x将按如下经验公式修正为x'：x' = ax^2 + bx + c，其中 a, b, c是已知的系数（a < 0）。
　　作为部队统帅，现在你要为这支部队进行编队，使得所有特别行动队修正后战斗力之和最大。试求出这个最大和。
　　例如， 你有4名士兵， x1 = 2, x2 = 2, x3 = 3, x4 = 4。经验公式中的参数为 a = –1, b = 10, c = –20。此时，最佳方案是将士兵组成 3个特别行动队：第一队包含士兵1和士兵2，第二队包含士兵3，第三队包含士兵 4。特别行动队的初始战斗力分别为4, 3, 4，修正后的战斗力分别为 4, 1, 4。修正后的战斗力和为 9，没有其它方案能使修正后的战斗力和更大。 

【输入文件】

    输入由三行组成。第一行包含一个整数 n，表示士兵的总数。第二行包含三个整数 a,  b,  c，经验公式中各项的系数。第三行包含 n 个用空格分隔的整数 x1, x2, …, xn，分别表示编号为1, 2, …, n的士兵的初始战斗力。 

【输出文件】

输出一个整数，表示所有特别行动队修正后战斗力之和的最大值。

【输入样例】

 4 

 -1 10 -20

 2 2 3 4 

【输出样例】

9

【数据范围】

20%的数据中，n ≤ 1000；
50%的数据中，n ≤ 10,000；
100%的数据中，1  ≤  n  ≤  1,000,000，–5  ≤  a  ≤  –1，|b|  ≤  10,000,000，|c|  ≤
10,000,000，1 ≤ xi ≤ 100。 

【题目分析】

首先看到题目就可以知道这是一道动态规划，然而朴素的n^2算法只能拿到20分。

首先，对于这个题来说，其决策是单调的，这样我们就可以用单调性把时间复杂度优化到nlogn，实践证明这样可以过7~8个点。

然后决策单调了，我们就会想到斜率……

设对于f[i]，由k决策更新比j决策更优，且k<j，那么有如下式子(si表示前i项和）

F[j]+a*(si-sj)^2+b*(si-sj)+c<f[k]+a*(si-sk)^2+b*(si-sk)+c

转化过程略，最后可以化为如下式子：

2*a*si>[(s*sj^2-b*sj+f[j])-(a*sk^2-b*sk+f[k])]/(sj-sk)

设g[i]=a*si^2-b*si+f[i]，那么我们就可以把si作为横坐标，g[i]作为纵坐标来表示斜率，这样就可以维护一个凸形的曲线，每次把队首斜率小于2*a*si的点出队，用队首的点计算f[i]和g[i]，再用g[i]去维护队列曲线的凸形。

【代码实现】

 

Code

 1 program test1;
 2 var s,f,g:array[0..1000000]of int64;
 3     d:array[0..1000000]of longint;
 4     t,w,i,j,n:longint;
 5     a,b,c:int64;
 6 function xl(i,j:longint):extended;
 7 begin
 8     exit((g[j]-g[i])/(s[j]-s[i]));
 9 end;
10 begin
11     readln(n);
12     readln(a,b,c);
13     for i:=1 to n do
14     begin
15         read(s[i]);
16         inc(s[i],s[i-1]);
17     end;
18     t:=0;w:=0;d[0]:=0;g[0]:=0;
19     for i:=1 to n do
20     begin
21         while (t<w)and(xl(d[t],d[t+1])>=2*a*s[i]) do inc(t);
22         j:=d[t];
23         f[i]:=f[j]+a*sqr(s[i]-s[j])+b*(s[i]-s[j])+c;
24         g[i]:=a*sqr(s[i])-b*s[i]+f[i];
25         while (w>t)and(xl(d[w],i)>xl(d[w-1],d[w])) do dec(w);
26         inc(w);
27         d[w]:=i;
28     end;
29     writeln(f[n]);
30 end.

 

 

 

转载于:https://www.cnblogs.com/whitecloth/archive/2012/05/05/2484836.html