不知道题面中的代码是什么逻辑。

但是可以通过尝试,发现只要 n20182017n\ge20182017 时的输出均为 2018201720182017,据此模拟这个过程即可。

(只需要把题面中给的代码加上一个特判,时间复杂度即为正确的)

#include<iostream>
using namespace std;
long long f(long long n)
{
    if (n < 20180001) return n + 2017;
    return f(f(n - 2018));
}
int main()
{
    long long n;
    cin >> n;
    if (n >= 20182017) cout << 20182017;
    else cout << f(n) << endl;
    return 0;
}