算法与数据结构实验题 4.2 残忍的沟里学姐
★实验任务
现在沟里学姐有一把激光刀,现在有 n 个人站在一排(每隔一米站一个人), 沟里对每个人都有一个仇恨值 ai (仇恨值可正可负,因为有些人可能是他的队友, 可能是他的敌人) ,沟里为了消灭更多敌人,因此他会不择手段,可能也把自己队友给杀了。他开始用他激光刀向右刺杀(只能进行一次刺杀),他想让他刺杀的人的仇恨值和最大。沟里的激光刀的长度最多可调节到 k 米,请你帮他计算他发动激光刀的位置 l 和激光刀结束的为 r (r-l+1<=k)。当存在多种方案使得仇恨值最大时,选择发动位置 l 最小的方案,再选择激光刀调节长度 r-l+1 最小的方案。
★数据输入
输入第一行为两个正整数 n(1<=n<=100000),k(0<k<=n),表示出现n 个人, 沟里的激光刀最多调节到 k 米。
输入第二行为 n 个整数,表示沟里对在 i 米处的人的仇恨值
ai(-1000<=a[i]<=1000)。
★数据输出
输出为一行三个数。第一个数为可获得的最大仇恨值和,第二个数为激光刀发位置 l,第三个数为激光刀结位置 r。
输入示例 1 | 输出示例 1 |
6 3 -1 2 -6 5 -5 6 | 6 4 6 |
输入示例 | 输出示例 |
5 5 -1 -1 -1 -1 -1 | -1 1 1 |
★hint
样例 1:沟里站在 4 米的地方发出两米长的激光刀,刺杀了 a[4]、a[5]、a[6]。
解题思路:
滑窗算法。。。
#include <cstdio>
#include <map>
#include <queue>
using namespace std;
pair <int ,int> p;
int main()
{
int n,sum=0,maxsum=-100000,l,r,k,x;
scanf ("%d %d",&n,&k);
//初始化一个队列 pair<int,int>储存元素值,元素位置 。
queue <pair<int,int> > q;
for (int i=0;i<n;i++)
{
scanf ("%d",&x);
while ((q.size()>=k)||(q.size()>1&&q.front().first<0))
{// 当队列长度大于等于K时,弹出队首。
// 当队首元素小于0,且队列长度大于1时,弹出对手元素。
sum=sum-q.front().first;
q.pop();
if (sum>maxsum)
{
maxsum=sum;
l=q.front().second+1;
r=q.back().second+1;
}
}
p.first=x;
p.second=i;
q.push(p);
while (sum<0)
{//当当前子序列和小于零时,清零,重新计算子序列和。
sum-=q.front().first;
q.pop();
}
sum+=x;
if (q.size()<=k&&sum>maxsum)
{
maxsum=sum;
l=q.front().second+1;
r=q.back().second+1;
}
}
printf ("%d %d %d\n",maxsum,l,r);
return 0;
}