c++的泡面问题求解！

Question

泡面是机房中的同学们的通用货币标配食物，L老师经常用它奖励同学们。今天，L老师又奖励同学们泡面了。
机房中的电脑有 xx 行 xx 列，电脑的个数刚好等于机房中人的总数。L老师有一个装满泡面的正方形盒子，里面装了 ({x-1})^{2}(x−1) 2 盒泡面，提示：({x-1})^{2} = x^2-2x+1(x−1) 2 =x 2−2x+1。L老师现在知道有kk个可怜的人没拿到泡面，他想知道机房里一列有几个电脑。

【输入格式】
共一行一个数，即kk

【输出格式】
一个数，即xx

输入数据 1
7
输出数据 1
4

Answer 1

#include<bits/stdc++.h>
#define int long long
#define PII pair<int,int>
using namespace std;
const int inf = 0x3f3f3f3f3f3f3f3f, N = 1e6 + 5, mod = 1e9 + 7;
signed main()
{
    ios_base::sync_with_stdio(0); cin.tie(0), cout.tie(0);
    int k;
    cin >> k;
    for (int i = 1; i <= N; i++) {
        if (i  * i  - (i-1) * (i-1) == k) {
            cout << i << '\n';
            break;
        }
    }
}