leetcode2834--找出美丽数组的最小和

_不会dp不改名_

于 2024-03-08 17:53:44 发布

阅读量801

点赞数 19

分类专栏： # 贪心文章标签：算法

本文链接：https://blog.csdn.net/bdn_nbd/article/details/136568211

版权

贪心专栏收录该内容

3 篇文章 0 订阅

订阅专栏

文章介绍了如何使用贪心策略构造一个满足特定条件的序列，以求得其最小和。当n小于目标值的一半时，按顺序累加；否则，从目标值开始逆序累加。代码中给出了Solution类的minimumPossibleSum函数实现。

摘要由CSDN通过智能技术生成

1. 题意

求一个序列和。序列 $a$ 满足：

大小为 $n$
$\forall 0\le i,j \lt n,i \ne j,a_i+a_j \ne target$

找出美丽数组的最小和

2. 题解

贪心的构造这个序列。

当 $\le target/2$ 时，依次累加序列即可。

当 $n > t a r g e t /2$ 时，从 $t a r g e t$ 开始累加。

$m=target/2\\ ans= \begin{cases} n(n+1)/2 \quad n \le m\\ m(m+1)/2 +(target+target+n-m-1)(n-m)/2\quad n \gt m \end{cases}$

代码

class Solution {
public:
    int minimumPossibleSum(int n, int target) {
        long long mid = target/2;
        long long MOD = 1e9+7;

        long long ans = 0;
        if (n <= mid) {
            ans = n % MOD * ((n + 1)%MOD)/2;
        }
        else {
            ans = (mid+1)*mid /2;
            ans %= MOD;

            long long cnt = n - mid;
            ans += (target + (target + cnt - 1)) * (cnt) /2; 
        }

        // int idx = 1;
        // int ans = 0;
        // for (int i = 0; i < n; ++i,++idx) {
        //     if ( i == mid )
        //         idx = target;
        //     ans += idx % MOD;
        //     ans %= MOD;
        // }


        return (int) (ans % MOD); 
    }
};