参考答案：

#include<iostream>
using namespace std;
int k, res;
int cnt = 0;
int main()
{
    cin >> k;
    for (int i = 1; ; ++i)
    {
        for (int j = i; j > 0; --j)
        {
            ++cnt;
            res += i;
            if (cnt == k)
            {
                break;
            }
        }
        if (cnt == k)
        {
            break;
        }
    }
    cout << res << endl;
    return 0;
}