一、题目
2485. 找出中枢整数 - 力扣(Leetcode)
给你一个正整数 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
二、题目解读
中枢整数
x
:
1
和x
之间的所有元素之和等于x
和n
之间所有元素之和。我们可以直接算出1到n的总和 sum=(1+n)*n/2,然后遍历1到x计算总和s=(1+x)*x/2,并与sum比较大小
sum-s==s 推出 √(sum)==s
也就是说我们可以直接计算sum开根号是否为整数来进行判断中枢整数
x
的存在
三、代码
java
Ⅰ、
class Solution {
public int pivotInteger(int n) {
int sum=(n+1)*n/2;
int s=0;
for (int i=1;i<=n;i++){
s+=i;
if (s==sum){
return i;
}else if (s>sum)
return -1;
sum-=i;
}
return -1;
}
}
Ⅱ、
class Solution {
public int pivotInteger(int n) {
int sum=(n+1)*n/2;
int x=(int) Math.sqrt(sum);
return x*x==sum?x:-1;
}
}
Python
Ⅰ、
class Solution(object):
def pivotInteger(self, n):
Sum = n * (n + 1) / 2
s = 0
for i in range(n + 1):
s += i
if s == Sum:
return i
Sum -= i
return -1
Ⅱ、
class Solution(object):
def pivotInteger(self, n):
s = (n * n + n) // 2
x = int(s ** 0.5)
if x * x == s:
return x
return -1