답안 #503261

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
503261 2022-01-07T15:11:44 Z Victor Necklace (Subtask 1-3) (BOI19_necklace1) C++17
9 / 85
465 ms 616 KB
// #pragma GCC target ("avx,avx2,fma")
// #pragma GCC optimize ("Ofast,inline") // O1 - O2 - O3 - Os - Ofast
// #pragma GCC optimize ("unroll-loops")
#include <bits/stdc++.h>

using namespace std;

#define rep(i, a, b) for (int i = (a); i < (b); ++i)
#define per(i, a, b) for (int i = (b - 1); i >= (a); --i)
#define trav(a, x) for (auto &a : x)

#define all(x) x.begin(), x.end()
#define sz(x) x.size()
#define pb push_back
#define debug(x) cout << #x << " = " << x << endl

#define umap unordered_map
#define uset unordered_set

typedef pair<int, int> ii;
typedef pair<int, ii> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef vector<vi> vvi;

typedef long long ll;
typedef pair<ll, ll> pll;
typedef vector<ll> vll;
typedef vector<pll> vpll;

const ll INF = 1'000'000'007, MAXN = 3000;
const ll num_bases = 1, bases[] = {31};
ll base_pow[num_bases][MAXN];

string jill, jane;

int main() {
    cin.tie(0)->sync_with_stdio(0);
    cin.exceptions(cin.failbit);

    rep(i, 0, num_bases) base_pow[i][0] = 1;
    rep(i, 0, num_bases) rep(j, 1, MAXN) base_pow[i][j] = base_pow[i][j - 1] * bases[i] % INF;

    cin >> jill >> jane;

    umap<int, int> possible;
    int n = min(sz(jill), sz(jane));

    int lo = 0, hi = n;
    int start_jill = 0, start_jane = 0;
    while (lo != hi) {
        int mid = (lo + hi + 1) / 2;

        ll hash_val = 0;
        rep(i, 0, mid - 1) hash_val = (hash_val * bases[0] + jill[i]) % INF;
        rep(i, mid - 1, sz(jill)) {
            hash_val = (hash_val * bases[0] + jill[i]) % INF;
            possible.emplace(hash_val, i - mid + 1);
            hash_val = ((hash_val - base_pow[0][mid - 1] * jill[i - mid + 1]) % INF + INF) % INF;
        }

        int sjill = -1, sjane = -1;
        rep(k, 0, 2) {
            hash_val = 0;
            rep(i, 0, mid - 1) hash_val = (hash_val * bases[0] + jane[i]) % INF;
            rep(i, mid - 1, sz(jane)) {
                hash_val = (hash_val * bases[0] + jane[i]) % INF;

                ll tmp_val = hash_val;
                rep(j, 0, mid) {
                    if (possible.count(tmp_val)) {
                        if (k) {
                            sjane = i - mid + 1;
                            sjane = sz(jane) - sjane;
                            sjane -= mid;
                        } else
                            sjane = i - mid + 1;
                        sjill = possible[tmp_val];
                    }

                    tmp_val = ((tmp_val - base_pow[0][mid - 1] * jane[i - mid + 1 + j]) % INF + INF) % INF;
                    tmp_val = (tmp_val * bases[0] + jane[i - mid + 1 + j]) % INF;
                }

                hash_val = ((hash_val - base_pow[0][mid - 1] * jane[i - mid + 1]) % INF + INF) % INF;
            }
            reverse(all(jane));
        }

        if (sjill == -1)
            hi = mid - 1;
        else {
            lo = mid;
            start_jill = sjill;
            start_jane = sjane;
        }
        possible.clear();
    }

    string s1 = jill.substr(start_jill, lo);
    string s2 = jane.substr(start_jane, lo);
    bool yes = 0;
    rep(k, 0, 2) {
        string cur = s2 + s2;
        rep(i, 0, lo) if (s1 == cur.substr(i, lo)) yes = 1;
        reverse(all(s2));
    }
    if (!yes) assert(0);
    cout << lo << endl;
    cout << start_jill << ' ' << start_jane << endl;

    return 0;
}

Compilation message

necklace.cpp: In function 'int main()':
necklace.cpp:8:42: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    8 | #define rep(i, a, b) for (int i = (a); i < (b); ++i)
      |                                          ^
necklace.cpp:56:9: note: in expansion of macro 'rep'
   56 |         rep(i, mid - 1, sz(jill)) {
      |         ^~~
necklace.cpp:8:42: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    8 | #define rep(i, a, b) for (int i = (a); i < (b); ++i)
      |                                          ^
necklace.cpp:66:13: note: in expansion of macro 'rep'
   66 |             rep(i, mid - 1, sz(jane)) {
      |             ^~~
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 332 KB Output is correct
2 Partially correct 1 ms 332 KB Output is partially correct
3 Correct 1 ms 336 KB Output is correct
4 Partially correct 1 ms 332 KB Output is partially correct
5 Correct 1 ms 332 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 332 KB Output is correct
2 Partially correct 1 ms 332 KB Output is partially correct
3 Correct 1 ms 336 KB Output is correct
4 Partially correct 1 ms 332 KB Output is partially correct
5 Correct 1 ms 332 KB Output is correct
6 Correct 11 ms 332 KB Output is correct
7 Partially correct 16 ms 336 KB Output is partially correct
8 Correct 9 ms 340 KB Output is correct
9 Partially correct 13 ms 352 KB Output is partially correct
10 Partially correct 13 ms 352 KB Output is partially correct
11 Partially correct 9 ms 336 KB Output is partially correct
12 Correct 11 ms 348 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 332 KB Output is correct
2 Partially correct 1 ms 332 KB Output is partially correct
3 Correct 1 ms 336 KB Output is correct
4 Partially correct 1 ms 332 KB Output is partially correct
5 Correct 1 ms 332 KB Output is correct
6 Correct 11 ms 332 KB Output is correct
7 Partially correct 16 ms 336 KB Output is partially correct
8 Correct 9 ms 340 KB Output is correct
9 Partially correct 13 ms 352 KB Output is partially correct
10 Partially correct 13 ms 352 KB Output is partially correct
11 Partially correct 9 ms 336 KB Output is partially correct
12 Correct 11 ms 348 KB Output is correct
13 Runtime error 465 ms 616 KB Execution killed with signal 6
14 Halted 0 ms 0 KB -