cf1811D - Umka and a Long Flight(思维)

传送门

#include <bits/stdc++.h>

using namespace std;

#define endl "\n"
#define debug cout<<"debug"<<endl
mt19937 rd(time(0));
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<int, int> PII;
const double eps = 1e-5;
const double PI = 3.14159265358979323;
const int N = 3e5+10, M = 2*N, mod = 1e9+7;
const int INF = 0x3f3f3f3f;

int n, x, y;
int f[N];

void init()
{
    f[0] = f[1] = 1;
    for(int i = 2; i<=50; i++) f[i] = f[i-1] + f[i-2];
}

// 结论: f[n]*f[n+1] = f[0]*f[0] + f[1]*f[1] + ... + f[n]*f[n]
// 即若要将一个f[n+1] f[n]长宽的矩形分成n+1个变成为斐波那契的矩形, 实际就是分成边长为 f[0] f[1] ... f[n]的正方形
// 染色方块为f[0] , 那么每次贪心按大的分, 看染色方块的位置, 如果当前矩形, 染色方块能够分出一个f[n]的正方形就递归往下分
// 否则就不存在答案, 因为长f[n+1] > 宽f[n] 所以每次看y的位置
// 若y>f[n], 即y的左边存在一个f[n]的正方形那么就递归分割, y的位置递归为f[n+1]-y+1
// 若y<f[n-1], 因为f[n+1] = f[n] + f[n-1], 所以y的右边存在, y的位置递归为y
// 若都不满足, 即当前不能分出一个f[n]的正方形, 返回false

void solve()
{
    cin>>n>>x>>y;
    function<bool(int, int, int)> dfs = [&](int n, int x, int y) -> bool
    {
        if(!n) return 1;
        if(y>f[n]) return dfs(n-1, f[n+1]-y+1, x); //
        else if(y<=f[n-1]) return dfs(n-1, y, x);
        else return 0;
    };
    cout<<(dfs(n, x, y) ? "YES" : "NO")<<endl;
}

int main()
{
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    init();
    int T;
    T = 1;
    cin>>T;
    while(T -- )
    {
        solve();
    }
    return 0;
}