不知道题面中的代码是什么逻辑。
但是可以通过尝试,发现只要 时的输出均为 ,据此模拟这个过程即可。
(只需要把题面中给的代码加上一个特判,时间复杂度即为正确的)
#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;
}