备战 清华大学 上机编程考试-冲刺前50%,倒数第6天

真题训练:

T1:舞蹈团 - 排序+滑动窗口

生活在在外星球X上的小Z想要找一些小朋友组成一个舞蹈团,于是他在网上发布了信息,一共有 \(n\) 个人报名面试。面试必须按照报名的顺序 依次进行。小Z可以选择在面试完若干小朋友以后,在所有已经面试过 的小朋友中进行任意顺序的挑选,以组合成一个舞蹈团。虽然说是小朋友,但是外星球X上的生态环境和地球上的不太一样,这些小朋友的身高可能相差很大。小Z希望组建的这个舞蹈团要求至少有 \(m\) 个小朋友,并且这些小朋友的最高身高和最低身高之差不能超过 \(k\) 个长度单位。现在知道了这些小朋友的身高信息,问小Z至少要面试多少小朋友才能在已经面试过的小朋友中选出不少于 \(m\) 个组成舞蹈团。

链接:REKCARC-TSC-UHT/大三小学期/保研考试/推研题目/THUPUB2017/statements/tuoj/day1/interview.md at master · PKUanonym/REKCARC-TSC-UHT (github.com)

解:

(在"轩轩"前辈的代码上,做了一些优化)

#include<iostream>
#include<vector>
#include<algorithm>
#include<stdio.h>
#include<stdlib.h>
#include<string>

using namespace std;

//定义学生类
struct student
{
    int id;
    int h;
};

//重载小于符号:
bool operator <(const student& a,const student& b)
{
    return a.h < b.h;
}

//利用 插入排序 进行改进:
//这一次不再调用sort(vec.begin()+1,vec.begin()+m+i+1)
//但是,要实现相同的功能,只是内部实现必须更加高效
//前m-1个调用sort
//后面就调用mySort
void mySort(vector<student> &vec,int m,int i)
{
    //vec[1] - vec[m+i-1]都是有序的,只要把vec[m+i]放到合适的位置即可:
    student tmp = vec[m+i];
    for(int u = m+i-1 ; u>=1 ; u--)
    {
        if(vec[u] < tmp)
        {
            //把tmp放到u+1的位置,把u+1 - m+i-1后移1个位置
            for(int w = m+i-1; w>=u+1 ;w--)
            {
                vec[w+1] = vec[w];
            }
            vec[u+1] = tmp;
            break;
        }
        else if(u == 1 && vec[u].h > tmp.h)
        {
            //tmp需要放到第一个位置
            for(int w = m+i-1 ; w>=1;w--)
            {
                vec[w+1] = vec[w];
            }
            vec[1] = tmp;
            break;
        }
    }
}

int main()
{
    //处理输入:
    int n,m,k;
    cin>>n>>m>>k;
    //输入n个小朋友的身高
    vector<student> vec(n+1);

    for(int i = 1; i<=n;i++)
    {
        cin>>vec[i].h;
        vec[i].id = i; //id都是从1开始的
    }

    //从1到n一次遍历,选出 m个朋友, 最高 - 最矮  <= k
    //输出面试完 atleast 个人:

    //其实这个问题挺难的,因为,你要 选择合适的 m个人

    int atleast = 0;    //至少需要的面试的人数

    //--大概按照"轩轩"前辈的思路进行求解,但是有所改进
    sort(vec.begin()+1,vec.begin()+m);
    for(int i = 0 ;i<= n-m;i++)
    {
        mySort(vec, m,i) ; 
        //sort(vec.begin()+1,vec.begin()+m+i+1); //sort是左闭右开的
        for(int j = 1;j<=i+1 ; j++)          //滑动窗口
        {
            //检查是否满足 最大 最小的 差值 <=k的条件:
            if(vec[j+m-1].h - vec[j].h <=k)
            {
                //搞定:
                cout<<m+i<<endl;
                return 0;
            }
        }
    }

    cout<<"impossible"<<endl;



    //真的很像一个dp动态规划:

    //那么,怎么设置初始状态 和 转移方程呢?

    //状态dp[i] : 至少面试了i人,才能获取到的满足 max - min <= k 的最大人数
    //初始状态dp[1] = 1
    //转移方程:
    
    //思路二: 每次加入一个人,都进行一次sort排序:
    //又有一些滑动窗口的感觉

    //我先写一个朴素的方法:

    //反正只是需要每次检查第i个,只要知道加入了第i个
    //是否满足即可:
    //这种朴素的方法,存在可以改进的地方,是否可以将前一次的信息利用起来

    //其中一种做法是,直接暴力:
    //从头m个,m+1个,m+2个,一直到m+(n-m)个,
    //每次进行sort排序,从头开始 检查, 窗口大小为m的是否满足

    //虽然"轩轩的代码挺不错的",但是
    //我觉得有一点可以改进,就是哪个for(int y = j,y<j+m;y++)找最大id这个循环
    //是不需要的,因为随着i的逐渐增大,tall一定是(m-1),m,m+1
    //"否则,我干嘛还要加入后一个元素"------请仔细思考这句话

    //还有一个可以改进的点,其实不必要每次全部sort,写一个 插入排序或许更优
    //毕竟前面都是排序好的    


    return 0 ;
}
相关推荐
pianmian14 小时前
python数据结构基础(7)
数据结构·算法
好奇龙猫6 小时前
【学习AI-相关路程-mnist手写数字分类-win-硬件:windows-自我学习AI-实验步骤-全连接神经网络(BPnetwork)-操作流程(3) 】
人工智能·算法
sp_fyf_20247 小时前
计算机前沿技术-人工智能算法-大语言模型-最新研究进展-2024-11-01
人工智能·深度学习·神经网络·算法·机器学习·语言模型·数据挖掘
香菜大丸7 小时前
链表的归并排序
数据结构·算法·链表
jrrz08287 小时前
LeetCode 热题100(七)【链表】(1)
数据结构·c++·算法·leetcode·链表
oliveira-time7 小时前
golang学习2
算法
南宫生8 小时前
贪心算法习题其四【力扣】【算法学习day.21】
学习·算法·leetcode·链表·贪心算法
懒惰才能让科技进步9 小时前
从零学习大模型(十二)-----基于梯度的重要性剪枝(Gradient-based Pruning)
人工智能·深度学习·学习·算法·chatgpt·transformer·剪枝
Ni-Guvara9 小时前
函数对象笔记
c++·算法
泉崎10 小时前
11.7比赛总结
数据结构·算法