Atcoder1371题解

我们先分析一下题目。

假设摆了 $k$ 行，可以进行如下推导：

$k\times (k+1)÷2=n$

$k\times (k+1)=n\times 2$

因为 $k^2<k\times (k+1)<(k+1{)}^2$，也即 $k^2<2\times n<(k+1{)}^2$ ，开方可以得到 $k<\sqrt{2\times n}<k+1$ ，说明 $\lfloor \sqrt{2\times n}\rfloor =k$ 。

所以可以通过判断 $\lfloor \sqrt{2\times n}\rfloor \times (\lfloor \sqrt{2\times n}\rfloor +1)$ 是否等于 $n\times 2$ 即可。

上代码：

#include <bits/stdc++.h>
using namespace std;
long long n,k;
int main(){
	cin>>n;
	k=sqrt(n*2);
	if(k*(k+1)/2==n)cout<<k;
	else cout<<-1;
        return 0;
}