結果

問題 No.2402 Dirty Stairs and Shoes
ユーザー nystnyst
提出日時 2023-08-04 22:22:22
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 155 ms / 2,000 ms
コード長 1,739 bytes
コンパイル時間 3,854 ms
コンパイル使用メモリ 200,648 KB
実行使用メモリ 30,744 KB
最終ジャッジ日時 2024-12-20 02:42:12
合計ジャッジ時間 6,696 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <iomanip>
#include <algorithm>
#include <map>
#include <cmath>
#include <bitset>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <tuple>
#include <atcoder/all>
#include <sstream> // std::stringstream
#include <regex>
#include <fstream>
using namespace atcoder;
using mint = modint998244353;
//using mint = modint1000000007;
using namespace std;
using ll = long long;
static const double pi = 3.141592653589793;
const int INF = (1 << 29);
const ll mod = 998244353;
// ifstream ifs(""); ファイルはワークスペースのトップにおく

int main() {

    int n,k;
    cin >> n >> k;
    int m1;cin >> m1;
    vector<int> a(m1);
    map<int,bool> mp;
    for(int i=0;i<m1;i++) {
        cin >> a[i];
        mp[a[i]] = true;
    }
    int m2;cin >> m2;
    vector<int> b(m2);
    map<int,bool> mp2;
    for(int i=0;i<m2;i++) {
        cin >> b[i];
        mp2[b[i]] = true;
    }
    vector dp(n+1,vector<int>(2,0));
    //dp[n][0 or 1] 靴が汚れているか

    dp[0][0] = true;
    for(int i=0;i<n;i++){
        if(dp[i][0]){
            if(i+1<=n) {
                if(mp[i+1]) dp[i+1][1] = true;
                else dp[i+1][0] = true;
            }
            if(i+k<=n){
                if(mp[i+k]) dp[i+k][1] = true;
                else dp[i+k][0] = true;
            }
        }
        if(dp[i][1]){
            if(i+1<=n){
                if(mp2[i+1]) dp[i+1][0] = true;
                else dp[i+1][1] = true;
            }
            if(i+k<=n){
                if(mp2[i+k]) dp[i+k][0] = true;
                else dp[i+k][1] = true;
            }
        }
    }

    if(dp[n][0]) cout << "Yes" << endl;
    else cout << "No" << endl;

}
0