题解 | 小芳的排列构造

小芳的排列构造

https://www.nowcoder.com/practice/5e14876bd7aa4ad5953a1fa0cd804a79

def solve():
    import sys
    # 读取输入:兼容多空格/换行分隔的输入格式,拆分后转为列表
    input = sys.stdin.read().split()
    n = int(input[0])  # 目标构造数组的长度
    k = int(input[1])  # 需要满足的核心数值约束
    
    # 处理n=1的特殊边界情况
    if n == 1:
        # 仅当k=2时合法,输出1;否则无解输出-1
        if k != 2:
            print(-1)
            return
        print(1)
        return
    
    # 计算k的合法取值范围:超出范围则无解
    mini = n * 3 - 1  # k的最小值约束
    maxi = n * (n + 3) // 2  # k的最大值约束
    if k < mini or k > maxi:
        print(-1)
        return
    
    # 初始化标记数组:vis[i]表示数字i是否已被选入核心序列(索引0未使用,对应1~n)
    vis = [False] * (n + 1)
    k -= n * 2  # 调整k值,将原始k转换为贪心筛选的目标值
    
    res = []  # 存储贪心筛选出的核心序列
    # 贪心策略:从n-1倒序遍历,筛选需要优先加入的数字
    i = n - 1
    while k > 0:  # 直到k被消耗完为止
        if k >= i:
            k -= i  # 消耗k值
            res.append(i)  # 将当前数字加入核心序列
            vis[i] = True  # 标记为已使用
        i -= 1  # 倒序递减
    
    # 反转核心序列:恢复正序输出(因之前是倒序筛选)
    res.reverse()
    
    # 构造输出列表:避免多次IO,提升效率
    output = []
    # 第一步:加入核心序列
    for x in res:
        output.append(str(x))
    
    # 第二步:补充1~n-1中未被使用的数字
    for i in range(1, n):
        if not vis[i]:
            output.append(str(i))
    
    # 第三步:最后添加数字n(固定在数组末尾)
    output.append(str(n))
    
    # 拼接所有元素并输出
    print(' '.join(output))

# 程序入口
if __name__ == "__main__":
    solve()

全部评论

相关推荐

01-28 16:12
中南大学 Java
几年前还没有chatgpt的时候,刷题真的是很痛苦。刷不出来只能看题解,题解有几个问题:第一个是每次看的写题解的人都不一样,很难有一个统一的思路;第二个也是最重要的是,题解只提供了作者自己的思路,但是没有办法告诉你你的思路哪里错了。其实很少有错误的思路,我只是需要被引导到正确的思路上面去。所以传统题解学习起来非常困难,每次做不出来难受,找题解更难受。但是现在chatgpt能做很多!它可以这样帮助你&nbsp;-1.&nbsp;可以直接按照你喜欢的语言生成各种解法的题解和分析复杂度。2.&nbsp;把题和你写的代码都发给它,它可以告诉你&nbsp;你的思路到底哪里有问题。有时候我发现我和题解非常接近,只是有一点点🤏想错了。只要改这一点点就是最优解。信心倍增。3.&nbsp;如果遇到不懂的题解可以一行一行询问为什么要这样写,chatgpt不会嫌你烦。有时候我觉得自己的range写错了,其实那样写也没错,只是chat老师的题解有一点优化,这个它都会讲清楚。4.&nbsp;它可以帮你找可以用同类型解法来做的题。然后它可以保持解法思路不变,用一个思路爽刷一个类型的题。如果题目之间思路又有变化,它会告诉你只有哪里变了,其他的地方还是老思路。5.&nbsp;它也可以直接帮你总结模板,易错点。经过chat老师的指导,我最大的改变是敢刷题了。之前刷题需要先找某一个人写的算法题repo,然后跟着某一个人他的思路刷他给的几个题。如果想写别的题,套用思路失败了,没有他的题解,也不知道到底哪里错了;看别人的题解,思路又乱了。这个问题在二分查找和dp类型的题里面特别常见。但是现在有chat老师,他会针对我的代码告诉我我哪里想错了,应该怎么做;还按照我写代码的习惯帮我总结了一套属于我的刷题模板。每天写题全是正反馈!
牛客981:不刷才是爽
AI时代的工作 VS 传...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务