找出中枢整数

102 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第8天,点击查看活动详情

题目描述

给你一个正整数 n ,找出满足下述条件的 中枢整数 x :

  • 1 和 x 之间的所有元素之和等于 x 和 n 之间所有元素之和。

返回中枢整数 **x 。如果不存在中枢整数,则返回 -1 。题目保证对于给定的输入,至多存在一个中枢整数。

示例 1:

输入: n = 8
输出: 6
解释: 6 是中枢整数,因为 1 + 2 + 3 + 4 + 5 + 6 = 6 + 7 + 8 = 21

示例 2:

输入: n = 1
输出: 1
解释: 1 是中枢整数,因为 1 = 1

示例 3:

输入: n = 4
输出: -1
解释: 可以证明不存在满足题目要求的整数。

 

提示:

  • 1 <= n <= 1000

思路分析

题目给我们一个正整数n,我们需要在n中找到满足条件的中枢整数x,x需要满足以下条件:

  • 1 和 x 之间的所有元素之和等于 x 和 n 之间所有元素之和。

这里我们需要先计算两个数列各自的元素和,因为这两个数列都是差值为1的连续元素,也就是说两个数组都是等差数列,所以我们可以使用小学四年级学过的等差数列的n项和公式来计算元素和,再判断两组数列的和是否相等即可。

等差数列前N项和公式:S=(A1+An)N/2

所以1 和 x 之间的所有元素之和为:(1 + x) * x / 2

x 和 n 之间所有元素之和为:(x + n) * (n - x + i) / 2

因为这里是要在1到n之间找到一个中枢整数,所以我们只需要从1到n遍历判断当前整数是否满足条件,满足条件即可返回当前整数作为结果,遍历完仍未找到则返回-1。

AC代码

完整AC代码如下:

/**
 * @param {number} n
 * @return {number}
 */
var pivotInteger = function(n) {
    for(let i = 1; i <= n; i++){
        if(i * (i + 1) / 2 == ((i + n) * (n - i + 1) / 2)) return i;
    }
    return -1;
};

说在后面

本人为算法业余爱好者,平时只是随着兴趣偶尔刷刷题,如果上面分享有错误的地方,欢迎指出,感激不尽。