結果

問題 No.3223 K-XOR Increasing Sequence
ユーザー Nauclhlt🪷
提出日時 2025-07-06 17:56:30
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 85 ms / 2,000 ms
コード長 4,594 bytes
コンパイル時間 2,091 ms
コンパイル使用メモリ 200,676 KB
実行使用メモリ 11,036 KB
最終ジャッジ日時 2025-07-06 17:58:01
合計ジャッジ時間 20,357 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 69
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

using ll = long long;

template <class _ForwardIterator>
void print(_ForwardIterator first, _ForwardIterator last)
{
    bool f = true;
    for (auto it = first; it != last; ++it)
    {
        if (!f)
        {
            cout << " ";
        }
        cout << *it;
        f = false;
    }

    cout << endl;
}

int main()
{
    cin.tie(nullptr);
    ios::sync_with_stdio(false);

    int N, K;
    cin >> N >> K;
    ll X, Y;
    cin >> X >> Y;

    // 解説の通りの場合分け
    if (Y == 0)
    {
        cout << "No" << endl;
        return 0;
    }
    else if (X == 0 && K >= 2)
    {
        if (N >= K + 3)
        {
            vector<ll> A(N);
            if (K % 2 == 0)
            {
                A[0] = 0;
                A[1] = 1;
                for (int i = 2; i < N - 1; i++)
                {
                    A[i] = 3;
                }
                A[N - 1] = Y;
            }
            else if (K % 3 == 0)
            {
                A[0] = 0;
                A[1] = 3;
                vector<ll> cycle = {1, 3, 2};
                for (int i = 2; i < N - 1; i++)
                {
                    A[i] = cycle[(i - 2) % 3];
                }
                A[N - 1] = Y;
            }
            else if (K % 3 == 1)
            {
                A[0] = 0;
                A[1] = 2;
                for (int i= 2; i < N - 1; i++)
                {
                    int pos = (i - 2) % K;
                    if (pos < 2)
                    {
                        A[i] = 2;
                    }
                    else if (pos >= K - 2)
                    {
                        A[i] = 3;
                    }
                    else
                    {
                        A[i] = ((pos - 2) % 3) + 1;
                    }
                }
                A[N - 1] = Y;
            }
            else if (K % 3 == 2)
            {
                A[0] = 0;
                A[1] = 2;
                for (int i = 2; i < N - 1; i++)
                {
                    int pos = (i - 2) % K;
                    if (pos < K - 2)
                    {
                        A[i] = pos % 3 + 1;
                    }
                    else
                    {
                        A[i] = 3;
                    }
                }
                A[N - 1] = Y;
            }
            
            
            cout << "Yes" << endl;
            print(A.begin(), A.end());
            return 0;
        }
        else if (N == K + 1)
        {
            vector<ll> A(N);
            A[N - 1] = Y;
            cout << "Yes" << endl;
            print(A.begin(), A.end());
            return 0;
        }
        else if (N == K + 2)
        {
            if (Y == 1)
            {
                cout << "No" << endl;
                return 0;
            }
            else
            {
                vector<ll> A(N);
                A[N - 2] = 1;
                A[N - 1] = Y;
                cout << "Yes" << endl;
                print(A.begin(), A.end());
                return 0;
            }
        }
    }
    else if (K == 1)
    {
        vector<ll> A(N);
        if (Y - X >= N - 1)
        {
            for (int i = 0; i < N; i++)
            {
                A[i] = X + i;
            }
            A[N - 1] = Y;
            cout << "Yes" << endl;
            print(A.begin(), A.end());
            return 0;
        }
        else
        {
            cout << "No" << endl;
            return 0;
        }
    }
    else
    {
        vector<int> A(N);
        vector<int> B(K);
        if (K % 2 == 0)
        {
            for (int i = 0; i < K; i++)
            {
                B[i] = X;
            }
        }
        else
        {
            for (int i = 0; i < K - 3; i++)
            {
                B[i] = X;
            }
            
            if (__popcount(X) == 1)
            {
                int p = 0;
                while ((X & (1L << p)) != 0) p++;
                B[K - 3] = X;
                B[K - 2] = X | (1L << p);
                B[K - 1] = 1L << p;
            }
            else
            {
                int p = 0;
                while ((X & (1L << p)) == 0) p++;
                B[K - 3] = X;
                B[K - 2] = X ^ (1 << p);
                B[K - 1] = 1L << p;
            }
        }

        for (int i = 0; i < N; i++)
        {
            A[i] = B[i % K];
        }
        A[N - 1] = Y;

        cout << "Yes" << endl;
        print(A.begin(), A.end());
        return 0;
    }
}
0