题目描述
阿明是一名推销员,他奉命到螺丝街推销他们公司的产品。螺丝街是一条死胡同,出口与入口是同一个,街道的一侧是围墙,另一侧是住户。螺丝街一共有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时,阿明最多积累的疲劳值。
输入输出样例
5 1 2 3 4 5 1 2 3 4 5
15 19 22 24 25
5 1 2 2 4 5 5 4 3 4 1
12 17 21 24 27
说明
【输入输出样例1说明】
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说明】
X=1:向住户4推销,往返走路的疲劳值为4+4,推销的疲劳值为4,总疲劳值4+4+4=12。
X=2:向住户1、4推销,往返走路的疲劳值为4+4,推销的疲劳值为5+4,总疲劳值4+4+5+4=17。
X=3:向住户1、2、4推销,往返走路的疲劳值为4+4,推销的疲劳值为5+4+4,总疲劳值4+4+5+4+4=21。
X=4:向住户1、2、3、4推销,往返走路的疲劳值为4+4,推销的疲劳值为5+4+3+4,总疲劳值4+4+5+4+3+4=24。或者向住户1、2、4、5推销,往返走路的疲劳值为5+5,推销的疲劳值为5+4+4+1,总疲劳值5+5+5+4+4+1=24。
X=5:向住户1、2、3、4、5推销,往返走路的疲劳值为5+5,推销的疲劳值为5+4+3+4+1,
总疲劳值5+5+5+4+3+4+1=27。
【数据说明】
对于20%的数据,1≤N≤20;
对于40%的数据,1≤N≤100;
对于60%的数据,1≤N≤1000;
对于100%的数据,1≤N≤100000。
哈哈,我才不会告诉你我第一次作这题,模拟了一天得了60分呢。
说正解吧,考虑贪心,对于选x个人,和选(x+1)个人,后一种一定是先选了x是的人,再选一个没有被选过的人。
没有被选过的人分两种情况,一种是要多走一段,另一种则不。
建两个堆,分别维护两种情况,时间复杂度是O(n*log(n))。
#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
const int N=100005;
int n,pos,ans,lst;
bool vis[N];
struct node
{
int dis,tir,num;
}a[N];
struct cmp1
{
bool operator()(node c,node d)
{
return c.tir<d.tir;
}
};
struct cmp2
{
bool operator()(node c,node d)
{
return c.tir+2*c.dis<d.tir+2*d.dis;
}
};
priority_queue<node,vector<node>,cmp1>q1;
priority_queue<node,vector<node>,cmp2>q2;
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i].dis);
for(int i=1;i<=n;i++)
scanf("%d",&a[i].tir);
for(int i=1;i<=n;i++)
a[i].num=i;
for(int i=1;i<=n;i++)
q2.push(a[i]);
for(int i=1;i<=n;i++)
{
int val1,val2;
while(!q1.empty()&&vis[q1.top().num])
q1.pop();
if(q1.empty())
val1=0;
else
val1=q1.top().tir;
while(!q2.empty()&&q2.top().num<=lst)
q2.pop();
if(q2.empty())
val2=0;
else
val2=q2.top().tir+2*(q2.top().dis-pos);
if(val1>=val2)
{
ans+=val1;
vis[q1.top().num]=1;
q1.pop();
}
else
{
ans+=val2;
vis[q2.top().num]=1;
for(int i=lst+1;i<=q2.top().num;i++)
q1.push(a[i]);
lst=q2.top().num;
pos=a[lst].dis;
q2.pop();
}
printf("%d\n",ans);
}
return 0;
}