题目:阿明是一名推销员,他奉命到螺丝街推销他们公司的产品。螺丝街是一条死胡同,出口与入口是同一个,街道的一侧是围墙,另一侧是住户。螺丝街一共有N家住户,第i家住户到入口的距离为Si米。由于同一栋房子里可以有多家住户,所以可能有多家住户与入口的距离相等。阿明会从入口进入,依次向螺丝街的X家住户推销产品,然后再原路走出去。
阿明每走1米就会积累1点疲劳值,向第i家住户推销产品会积累Ai点疲劳值。阿明是工作狂,他想知道,对于不同的X,在不走多余的路的前提下,他最多可以积累多少点疲劳值
输入格式:
第一行有一个正整数N,表示螺丝街住户的数量。
接下来的一行有N个正整数,其中第i个整数Si表示第i家住户到入口的距离。数据保证S1≤S2≤…≤Sn<10^8。
接下来的一行有N个正整数,其中第i个整数Ai表示向第i户住户推销产品会积累的疲劳值。数据保证Ai<10^3。
输出格式:
输出N行,每行一个正整数,第i行整数表示当X=i时,阿明最多积累的疲劳值。
【数据说明】
对于20%的数据,1≤N≤20;
对于40%的数据,1≤N≤100;
对于60%的数据,1≤N≤1000;
对于100%的数据,1≤N≤1000000。
NOIP之前一直以为NOIP不会考线段树(其实真的没考,这题正解是贪心但是我不会%%%),所以写T4的时候打完暴力根本就没想到写线段树。暴力思路为一重循环枚举推销多少家,第二重循环找出消耗疲劳值最大的,然后改为0,输出答案,效率O(n^2)。
暴力代码如下:
var
len,tired:array[0..1000000]of longint;
n,i,j,max,s,ans,now,t,sum:longint;
begin
readln(n);
for i:=1 to n do read(len[i]);
for i:=1 to n do read(tired[i]);
ans:=0;
max:=0;
for i:=1 to n do
begin
max:=-maxlongint;
for j:=1 to n do
begin
if j>now then s:=(2*len[j])-(2*sum)+tired[j]
else s:=tired[j];
if s>max then
begin
t:=j;
max:=s;
end;
end;
if t>now then
begin
now:=t;
sum:=len[now];
end;
len[t]:=0;tired[t]:=0;
inc(ans,max);
writeln(ans);
end;
end.
然后我们可以发现我们找最大值效率是O(n),那我们就用线段树来维护最大值就行了,效率O(log(n)),则总时间效率为O(nlog(n))。
这里简单说一下思路。建两棵线段树,对于现在走到的点,一棵维护左边的疲劳值最大值,也就是只需要维护推销的疲劳值最大值,一棵维护右边的疲劳值最大值,需要维护推销的疲劳值+走路的疲劳值。
代码如下(3kb):
type
point=record
max,delta,l,r,num:longint;
end;
var
len,tired:array[0..1000000]of int64;
i,j:longint;
n,max,ans,now,tt,t1,t2,max2,l,x,y:int64;
a:array[0..1000000,1..2]of point;
procedure insert(x,num,delta,t:longint);
begin
if a[x,t].l=a[x,t].r then a[x,t].max:=0
else
begin
if num<=(a[x,t].l+a[x,t].r)>>1 then insert(2*x,num,delta,t);
if num>(a[x,t].l+a[x,t].r)>>1 then insert(2*x+1,num,delta,t);
if a[2*x,t].max>a[2*x+1,t].max then
begin
a[x,t].max:=a[2*x,t].max;
a[x,t].num:=a[2*x,t].num;
end
else
begin
a[x,t].max:=a[2*x+1,t].max;
a[x,t].num:=a[2*x+1,t].num;
end;
end;
end;
procedure build(x,l,r,t:longint);
begin
a[x,t].l:=l;a[x,t].r:=r;
if l=r then
begin
a[x,t].num:=l;
read(a[x,t].max);
a[x,t].delta:=0;
if t=1 then len[l]:=a[x,t].max;
if t=2 then tired[l]:=a[x,t].max;
end
else
begin
build(2*x,l,(l+r)>>1,t);
build(2*x+1,((l+r)>>1)+1,r,t);
if a[2*x,t].max>a[2*x+1,t].max then
begin
a[x,t].max:=a[2*x,t].max;
a[x,t].num:=a[2*x,t].num;
end
else
begin
a[x,t].max:=a[2*x+1,t].max;
a[x,t].num:=a[2*x+1,t].num;
end;
a[x,t].delta:=0;
end;
end;
procedure update(x,l,r,delta,t:longint);
begin
if (l<=a[x,t].l)and(a[x,t].r<=r) then
inc(a[x,t].delta,delta)
else
begin
if l<=(a[x,t].l+a[x,t].r)>>1 then update(2*x,l,r,delta,t);
if r>(a[x,t].l+a[x,t].r)>>1 then update(2*x+1,l,r,delta,t);
if a[2*x,t].max+a[2*x,t].delta>a[2*x+1,t].max+a[2*x+1,t].delta then
begin
a[x,t].max:=a[2*x,t].max+a[2*x,t].delta;
a[x,t].num:=a[2*x,t].num;
end
else
begin
a[x,t].max:=a[2*x+1,t].max+a[2*x+1,t].delta;
a[x,t].num:=a[2*x+1,t].num;
end;
end;
end;
function query(x,l,r,h,t:longint):int64;
var
ret:int64;
begin
ret:=0;
if (l<=a[x,t].l)and(a[x,t].r<=r) then
begin
ret:=a[x,t].max+a[x,t].delta;
if ret>=h then
tt:=a[x,t].num;
end
else
begin
inc(a[2*x,t].delta,a[x,t].delta);
inc(a[2*x+1,t].delta,a[x,t].delta);
inc(a[x,t].max,a[x,t].delta);
a[x,t].delta:=0;
if l<=(a[x,t].l+a[x,t].r)>>1 then ret:=query(2*x,l,r,h,t);
if r>(a[x,t].l+a[x,t].r)>>1 then
if ret<query(2*x+1,l,r,ret,t) then ret:=query(2*x+1,l,r,ret,t);
end;
query:=ret;
end;
begin
readln(n);
build(1,1,n,1);
build(1,1,n,2);
for i:=1 to n do update(1,i,i,len[i]+tired[i],1);
ans:=0;
max:=0;
tt:=0;
for i:=1 to n do
begin
max:=-maxlongint;
if now>0 then max:=query(1,1,now,0,2);
t1:=tt;
max2:=-maxlongint;
if now<n then max2:=query(1,now,n,0,1);
if max<max2 then
begin
max:=max2;
t1:=tt;
end;
if t1>now then
begin
update(1,t1,n,-2*len[t1],1);
insert(1,t1,0,1);
now:=t1;
end;
insert(1,t1,0,2);
inc(ans,max);
writeln(ans);
end;
end.
以下为官方数据情况: