제출 #151869

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
151869		forestryks	Matching (CEOI11_mat)	C++14	63 / 100	1618 ms	65540 KiB

mat

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

using ll = long long;
using ld = long double;
using pii = pair<int, int>;
#define rep(i, n) for (int (i) = 0; (i) < (n); ++(i))
#define all(x) (x).begin(), (x).end()
#define FAST_IO ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
#define f first
#define s second

const int MAXN = 2e6 + 5;
int m, n;
int p[MAXN];
int a[MAXN];
int up[MAXN];
int down[MAXN];
int pf[MAXN];

void calc() {
    set<pii> s;
    fill(up, up + m, -1);
    fill(down, down + m, -1);
    for (int i = 0; i < m; ++i) {
        auto it = s.lower_bound({a[i], 0});
        if (it != s.end()) {
            up[i] = it->s;
        }
        if (it != s.begin()) {
            down[i] = prev(it)->s;
        }
        s.insert({a[i], i});
    }
}

int main() {
    FAST_IO;
    cin >> m >> n;
    rep(i, m) {
        cin >> p[i];
        p[i]--;
    }
    rep(i, m) {
        a[p[i]] = i;
    }
    a[m] = -1;
    rep(i, n) {
        cin >> a[m + 1 + i];
    }

    calc();

    pf[0] = 0;
    for (int i = 1; i < n + m + 1; ++i) {
        if (i == m) {
            pf[i] = 0;
            continue;
        }
        int j = pf[i - 1];
        while (j > 0) {
            if (j == m) {
                j = pf[j - 1];
                continue;
            }

            int u = up[j];
            int d = down[j];

            if ((d != -1 && a[i - j + d] > a[i]) || (u != -1 && a[i - j + u] < a[i])) {
                j = pf[j - 1];
                continue;
            }

            pf[i] = j + 1;
            break;
        }

        if (j == 0) pf[i] = 1;
    }

    // rep(i, n + m + 1) {
    //     cout << a[i] << ' ';
    // }
    // cout << endl;

    // rep(i, m) {
    //     cout << down[i] << ' ';
    // }
    // cout << endl;

    // rep(i, m) {
    //     cout << up[i] << ' ';
    // }
    // cout << endl;

    // rep(i, n + m + 1) {
    //     cout << pf[i] << ' ';
    // }
    // cout << endl;

    vector<int> ans;
    for (int i = m + 1 + m - 1; i < n + m + 1; ++i) {
        if (pf[i] == m) {
            ans.push_back(i - m + 1 - m - 1);
        }
    }

    cout << ans.size() << endl;
    for (auto it : ans) {
        cout << it + 1 << ' ';
    }
    cout << endl;
}

• Subtask 1: 9 / 9
• Subtask 2: 9 / 9
• Subtask 3: 9 / 9
• Subtask 4: 9 / 9
• Subtask 5: 9 / 9
• Subtask 6: 9 / 9
• Subtask 7: 9 / 9
• Subtask 8: 0 / 9
• Subtask 9: 0 / 8
• Subtask 10: 0 / 10
• Subtask 11: 0 / 10