poj3612

来源:互联网 发布:汽车估价软件 编辑:程序博客网 时间:2024/06/15 16:37

【题意】

有一列树,总共有n(n<=100000)棵,给定树的高度,要在相邻树和树之间架设电线,代价为c*(两树的高度差的绝对值),

但可以增高树的高度,需要代价为<增高高度的平方>

求最小代价

【输入】

多组数据

第一行为n、c

接下来一行为n个数,表示高度(高度小于等于100)

【输出】

对于每组数据输出一个数表示最小代价


dp

用f[i][j]表示第i棵树高度为j的时候的最小代价

很容易想到枚举相邻两棵树高度

这样复杂度是O(n*h*h)

会超时

从左往右扫描树的时候,首先做一个预处理,处理出来最小值,然后再用o(1)复杂度求解

答案或者中间值可能会溢出,所以需要用int64


program poj3612;const  thu:int64=1000;var  n,c,i,j,k,o,s,e:longint;  ans:int64;  h:array [0..100001] of longint;  f:array [0..1,0..101] of int64;  low,high:array [0..101] of int64;function min (a,b:int64):int64;begin  if a<b then exit(a)         else exit(b);end;function max (a,b:int64):int64;begin  if a>b then exit(a)         else exit(b);end;begin  while not seekeof do    begin      read(n,c);      for i:=1 to n do        read(h[i]);      if n<=1 then        begin          writeln(0);          continue;        end;      o:=1;      for i:=0 to 101 do        if i<h[1] then f[o,i]:=maxlongint*thu                  else f[o,i]:=(h[1]-i)*(h[1]-i);      ans:=maxlongint;      for i:=2 to n do        begin          o:=1-o;          fillchar(f[o],sizeof(f[o]),63);          fillchar(low,sizeof(low),63);          fillchar(high,sizeof(high),63);          for j:=h[i-1] to 100 do            low[j]:=min(low[j-1],f[1-o,j]-j*c);          for j:=100 downto h[i] do            high[j]:=min(high[j+1],f[1-o,j]+j*c);          ans:=f[o,0];          for j:=h[i] to 100 do            begin              f[o,j]:=min(low[j]+j*c,high[j]-j*c)+(j-h[i])*(j-h[i]);              if i=n then ans:=min(ans,f[o,j]);            end;        end;      writeln(ans);    end;end.