結果

問題 No.2402 Dirty Stairs and Shoes
ユーザー monnumonnu
提出日時 2023-08-04 21:38:52
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 54 ms / 2,000 ms
コード長 1,332 bytes
コンパイル時間 1,612 ms
コンパイル使用メモリ 172,364 KB
実行使用メモリ 14,016 KB
最終ジャッジ日時 2024-12-20 02:30:58
合計ジャッジ時間 3,008 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #

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

int main()
{
    int N, K;
    cin >> N >> K;
    vector<int> A(N + 1, 0);
    int M1;
    cin >> M1;
    for (int i = 0; i < M1; i++)
    {
        int a;
        cin >> a;
        A[a] = -1;
    }
    int M2;
    cin >> M2;
    for (int i = 0; i < M2; i++)
    {
        int b;
        cin >> b;
        A[b] = 1;
    }

    vector<vector<int>> dp(N + 1, vector<int>(2, 0));
    dp[0][1] = 1;
    for (int i = 0; i < N; i++)
    {
        if (A[i + 1] == 1)
        {
            dp[i + 1][1] |= dp[i][0] | dp[i][1];
        }
        else if (A[i + 1] == -1)
        {
            dp[i + 1][0] |= dp[i][0] | dp[i][1];
        }
        else
        {
            dp[i + 1][0] |= dp[i][0];
            dp[i + 1][1] |= dp[i][1];
        }
        if (i + K <= N)
        {
            if (A[i + K] == 1)
            {
                dp[i + K][1] |= dp[i][0] | dp[i][1];
            }
            else if (A[i + K] == -1)
            {
                dp[i + K][0] |= dp[i][0] | dp[i][1];
            }
            else
            {
                dp[i + K][0] |= dp[i][0];
                dp[i + K][1] |= dp[i][1];
            }
        }
    }

    if (dp[N][1])
    {
        cout << "Yes" << endl;
    }
    else
    {
        cout << "No" << endl;
    }
}
0