NOIP2015普及组
推销员
题目描述
阿明是一名推销员,他奉命到螺丝街推销他们公司的产品。螺丝街是一条死胡同,出口与入口是同一个,街道的一侧是围墙,另一侧是住户。螺丝街一共有N家住户,第i家住户到入口的距离为Si米。由于同一栋房子里可以有多家住户,所以可能有多家住户与入口的距离相等。阿明会从入口进入,依次向螺丝街的X家住户推销产品,然后再原路走出去。
阿明每走1米就会积累1点疲劳值,向第i家住户推销产品会积累Ai点疲劳值。阿明是工作狂,他想知道,对于不同的X,在不走多余的路的前提下,他最多可以积累多少点疲劳值。
输入
第一行有一个正整数N(1≤N≤100000),表示螺丝街住户的数量。
接下来的一行有N个正整数,其中第i个整数Si表示第i家住户到入口的距离。数据保证S1≤S2≤…≤Sn<108。
接下来的一行有N个正整数,其中第i个整数Ai表示向第i户住户推销产品会积累的疲劳值。数据保证Ai<103。
输出
输出N行,每行一个正整数,第i行整数表示当X=i时,阿明最多积累的疲劳值。
样例输入
5
1 2 3 4 5
1 2 3 4 5
样例输出
15
19
22
24
25
提示
X=1:向住户5推销,往返走路的疲劳值为5+5,推销的疲劳值为5,总疲劳值为15。
X=2:向住户4、5推销,往返走路的疲劳值为5+5,推销的疲劳值为4+5,总疲劳值为5+5+4+5=19。
X=3:向住户3、4、5推销,往返走路的疲劳值为5+5,推销的疲劳值3+4+5,总疲劳值为5+5+3+4+5=22。
X=4:向住户2、3、4、5推销,往返走路的疲劳值为5+5,推销的疲劳值2+3+4+5,总疲劳值5+5+2+3+4+5=24。
X=5:向住户1、2、3、4、5推销,往返走路的疲劳值为5+5,推销的疲劳值1+2+3+4+5,总疲劳值5+5+1+2+3+4+5=25.
思路
阿明最多积累的疲劳值必定是最大距离*2+各家的推销疲劳值,但通过排序实现算法会导致超时,因此用优先队列的方式以实现算法
代码实现
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long ll;
const int N=100005;
struct line
{
int v,s;
bool operator < (line a) const
{
return v<a.v;
}
}t[N],s;
priority_queue<line> q;
int sum,n;
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&t[i].s);
for(int i=1;i<=n;i++) scanf("%d",&t[i].v);
s.s=s.v=0;
q.push(s);
int now=0,maxx=0;
for(int i=1;i<=n;i++)
{
int next=now;
s=q.top();
maxx=s.v;
for(int j=now+1;j<=n;j++)
{
if(t[j].v+(t[j].s-t[now].s)*2>maxx)
{
maxx=t[j].v+(t[j].s-t[now].s)*2;
next=j;
}
}
t[next].v+=(t[next].s-t[now].s)*2;
if(next!=now) q.push(t[next]);
for(int j=now+1;j<next;j++) q.push(t[j]);
s=q.top();
sum+=s.v;
printf("%d\n",sum);
q.pop();
now=next;
}
return 0;
}