解决leetcode第3826题.最小分割分数问题

3826.最小分割分数

难度:困难

问题描述:

给你一个整数数组nums和一个整数k。

你的任务是将nums分割成恰好k个子数组,并返回所有有效分割方案中最小可能的分数。

一个分割方案的分数是其所有子数组值的总和。

子数组的值定义为sumArr*(sumArr+1)/2,其中sumArr是该子数组元素的总和。

子数组是数组中连续的非空元素序列。

示例1:

输入:nums=5,1,2,1,k=2

输出:25

解释:

我们必须将数组分割成k=2个子数组。一种最优方案是51,2,1

第一个子数组的sumArr=5,value=5×6/2=15。

第二个子数组的sumArr=1+2+1=4,value=4×5/2=10。

该分割方案的分数为15+10=25,这是可能的最小分数。

示例2:

输入:nums=1,2,3,4,k=1

输出:55

解释:

由于必须分割成k=1个子数组,所有元素都属于同一个子数组:1,2,3,4

该子数组的sumArr=1+2+3+4=10,value=10×11/2=55。

该分割方案的分数为55,这是可能的最小分数。

示例3:

输入:nums=1,1,1,k=3

输出:3

解释:

我们必须将数组分割成k=3个子数组。唯一的有效分割方案是1,1,1

每个子数组的sumArr=1,value=1×2/2=1。

该分割方案的分数为1+1+1=3,这是可能的最小分数。

提示:

1<=nums.length<=1000

1<=numsi<=104

1<=k<=nums.length

问题分析:

将nums数组分割成恰好k个子数组,则这k个子数组的长度和必然等nums的长度,把数组nums的长度设为n,这个问题就可以抽象为如何把n分成k个数之和,把各种分法找出之后,再将每一种分法解析为相应的子数组,并求出对应的分数,最后在其中找到最小分数即可。

程序如下:

python 复制代码
#把n分成k个数之和,返回各种分法
def get_division_method(n,k):
    if k==1:
        return [[n]]
    elif k==2:
        t=[]
        for i in range(1,n):
            t.append([i,n-i])
        return t
    else:
        t=[]
        for i in range(1,n):
            m=get_division_method(n-i,k-1)
            for j in m:
                j.append(i)
                t.append(j)
        return t

#根据一个分解方案解析生成对应的子数组并返回
def analysis_plan(nums,a):
    t=[]
    k=0
    for i in a:
        b=nums[k:k+i]
        t.append(b)
        k=k+i
    return t

#计算一个子数组的分数,并返回
def get_score(sub_array):
    s=sum(sub_array)
    return s*(s+1)/2

#主程序
nums=eval(input('pls input nums='))
k=int(input('pls input k='))
n=len(nums)
a=get_division_method(n,k)
score=[]
for i in a:
    t=analysis_plan(nums,i)
    s=0
    for j in t:
        s+=get_score(j)
    score.append([t,s])
score.sort(key=lambda x:x[1])
print(f'最优方案是{score[0][0]},该分割方案的分数为{int(score[0][1])}')

运行实例一

pls input nums=1,2,3

pls input k=1

最优方案是\[1, 2, 3],该分割方案的分数为21

运行实例二

pls input nums=1,2,3,4,5

pls input k=2

最优方案是\[1, 2, 3, 4, 5],该分割方案的分数为66

运行实例三

pls input nums=5,4,2,1

pls input k=3

最优方案是\[5, 4, 2, 1],该分割方案的分数为31

运行实例四

pls input nums=1,3,5,7,9

pls input k=4

最优方案是\[1, 3, 5, 7, 9],该分割方案的分数为98

相关推荐
专业程序开发源12 分钟前
springboot篮球联赛管理系统13635-计算机课程设计、毕业设计
vue.js·spring boot·后端·python·django·php·课程设计
lie..15 分钟前
30天从零开始学AI应用开发(Day 3):30 分钟搭好开发环境,Python + VSCode 一次配齐
人工智能·python·大模型
β添砖java15 分钟前
机器学习2 KNN算法、距离度量、特征预处理、超参数选择、手写数字、鸢尾花
人工智能·算法·机器学习
TAN-90°-22 分钟前
Deep Learning for Computer Vision——Generative Models 2
人工智能·深度学习·神经网络·算法·目标检测·机器学习·计算机视觉
Rabitebla30 分钟前
【Linux系统编程】指令(三):创建、删除、复制、移动、看内容
linux·数据结构·c++·算法
benchmark_cc32 分钟前
策略临时需要一批股票的最新价格,先查行情还是先建股票池?——从量化策略执行逻辑看数据获取顺序
python·数据分析·pandas·量化交易·股票数据·quantdash
成为深度学习高手33 分钟前
XLinear:用一个轻量 MLP,带外生变量做长期时间序列预测
python·深度学习·时序数据库
啊吧怪不啊吧1 小时前
LangChain之模型调用
python·langchain
溪语流沙1 小时前
Django + Vue电商项目第001讲:开篇|注册登录加增删改查,那不是电商
redis·python·mysql·docker·typescript·django·vue
夜雪一千1 小时前
如何使用BeautifulSoup库来提取HTML中的特定元素
python