結果
問題 | No.1508 Avoid being hit |
ユーザー |
👑 |
提出日時 | 2021-05-14 23:25:27 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,101 ms / 3,000 ms |
コード長 | 2,793 bytes |
コンパイル時間 | 1,538 ms |
コンパイル使用メモリ | 110,448 KB |
実行使用メモリ | 12,192 KB |
最終ジャッジ日時 | 2024-10-02 05:15:51 |
合計ジャッジ時間 | 20,784 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 43 |
ソースコード
#include <cassert> #include <cmath> #include <cstdint> #include <cstdio> #include <cstdlib> #include <cstring> #include <algorithm> #include <bitset> #include <complex> #include <deque> #include <functional> #include <iostream> #include <map> #include <numeric> #include <queue> #include <set> #include <sstream> #include <string> #include <unordered_map> #include <unordered_set> #include <utility> #include <vector> using namespace std; using Int = long long; template <class T1, class T2> ostream &operator<<(ostream &os, const pair<T1, T2> &a) { return os << "(" << a.first << ", " << a.second << ")"; }; template <class T> void pv(T a, T b) { for (T i = a; i != b; ++i) cerr << *i << " "; cerr << endl; } template <class T> bool chmin(T &t, const T &f) { if (t > f) { t = f; return true; } return false; } template <class T> bool chmax(T &t, const T &f) { if (t < f) { t = f; return true; } return false; } using BS = bitset<100010>; #ifdef LOCAL constexpr int BLOCK = 2; #else constexpr int BLOCK = 300; #endif int N, Q; vector<int> A, B; int main() { for (; ~scanf("%d%d", &N, &Q); ) { A.resize(Q); B.resize(Q); for (int i = 0; i < Q; ++i) { scanf("%d", &A[i]); } for (int i = 0; i < Q; ++i) { scanf("%d", &B[i]); } vector<int> is; for (int i = 0; i < Q; i += BLOCK) { is.push_back(i); } const int len = is.size(); is.push_back(Q); vector<BS> bss(len + 1); bss[0].reset(); for (int x = 1; x <= N; ++x) { bss[0][x] = true; } for (int j = 0; j < len; ++j) { BS bs = bss[j]; for (int i = is[j]; i < is[j + 1]; ++i) { bs = bs >> 1 | bs | bs << 1; bs[0] = false; bs[N + 1] = false; bs[A[i]] = false; bs[B[i]] = false; } bss[j + 1] = bs; } if (bss[len].any()) { vector<int> ans(Q + 1); ans[Q] = bss[len]._Find_first(); for (int j = len; --j >= 0; ) { vector<BS> css(is[j + 1] - is[j] + 1); css[0] = bss[j]; for (int i = is[j]; i < is[j + 1]; ++i) { const int pos = i - is[j]; css[pos + 1] = css[pos] >> 1 | css[pos] | css[pos] << 1; css[pos + 1][0] = false; css[pos + 1][N + 1] = false; css[pos + 1][A[i]] = false; css[pos + 1][B[i]] = false; } for (int i = is[j + 1]; --i >= is[j]; ) { const int pos = i - is[j]; for (int x = ans[i + 1] - 1; x <= ans[i + 1] + 1; ++x) { if (css[pos][x]) { ans[i] = x; break; } } assert(ans[i]); } } puts("YES"); for (int i = 0; i <= Q; ++i) { printf("%d\n", ans[i]); } } else { puts("NO"); } } return 0; }