牛客真题(8)-疯狂队列

今天继续刷牛客真题,给定一个队列,求一个疯狂队列,这个队列满足,两两相邻的元素差的绝对值的和最大。

分析:
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)

猜你喜欢

转载自blog.csdn.net/JerryZengZ/article/details/88870421