今天继续刷牛客真题,给定一个队列,求一个疯狂队列,这个队列满足,两两相邻的元素差的绝对值的和最大。
分析:
1、方法一:通过找到规律,首先对数组进行排序,然后取最大和最小,计算差,然后加上最大减去次小,以及次大减去最小的差,知道最后一个的时候,判断是与较大值进行求差还是与较小值求差,这样就可以得到疯狂队列的最大值。(C++实现)
2、方法二:通过简单的比较方法,首先对数组进行排序,然后取最大值和最小的值,然后判断四个值(最大,次大,最小,次小)之间的差,找到最大的差,将这个元素加入到新的列表中,然后更改最大或最小,依次遍历完整个数组,最后计算新的列表中两两元组差的绝对值的和,就是疯狂队列的值。
问题:
1、队列的插入,从头插入以及从尾部添加;
2、C++与Python中对数组的改变不同,C++对数组进行添加,需要移动元素,而Python对列表添加不需要移动。
附上C++代码:
#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
vector<int> tt;
int n;
int a;
cin>>n;
for (int i=0;i<n;i++)
{
cin>>a;
tt.push_back(a);
}
std::sort(tt.begin(),tt.end(),std::greater<int>());
//cout<<tt[0]<<endl;
int max = tt[0]; //获取最大值
int min = tt[n-1]; //获取最小值
int maxline = 1; //获取下一个最大值的下标
int minline = n-1-1; //获取下一个最小值的小标
int sum = max - min;
while (minline > maxline)
{
sum += max - tt[minline] + tt[maxline] - min;
max = tt[maxline++];
min = tt[minline--];
}
if ((max-tt[minline])>(tt[maxline]-min))
{
sum += max-tt[minline];
}
else
{
sum+= tt[maxline]-min;
}
cout<<sum<<endl;
return 0;
}
附上Python代码:
n = int(input())
l = list(map(int,input().split()))
l.sort()
temp =[l[0]]
left = 1
right = n-1
res = 0
while left<=right:
r0 = abs(temp[0] - l[right])
lr = abs(temp[-1] - l[right])
l0 = abs(temp[0]-l[left])
l1 = abs(temp[-1]-l[left])
tl = [r0,lr,l0,l1]
m = max(tl)
if tl.index(m)==0:
temp.insert(0,l[right])
right-=1
if tl.index(m)==1:
temp.append(l[right])
right-=1
if tl.index(m)==2:
temp.insert(0,l[left])
left+=1
if tl.index(m)==3:
temp.append(l[left])
left+=1
for i in range(n-1):
res+=abs(temp[i+1]-temp[i])
print(res)