Editorial for Bedao Regular Contest 03 - WORDGAME


Remember to use this editorial only when stuck, and not to copy-paste code from it. Please be respectful to the problem author and editorialist.
Submitting an official solution before solving the problem yourself is a bannable offence.

Author: bedao

Sử dụng ~1~ vòng lặp, tạo một xâu lưu trữ tạm thời chữ cái cuối của xâu vừa nhập vào sau đó so sánh với chữ cái đầu của xâu tiếp theo. Sử dụng map<string, long long> để kiểm tra xem xâu kí tự đã xuất hiện trước đó hay chưa.

Độ phức tạp: ~O(nlogn)~, có thể sử dụng unordered_map để có ~O(n)~

Code mẫu

/*#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("avx,avx2,fma")*/

#include <bits/stdc++.h>

#define for1(i,a,b) for (int i = a; i <= b; i++)
#define for2(i,a,b) for (int i = a; i >= b; i--)
#define int long long

#define sz(a) (int)a.size()
#define pii pair<int,int>
#define pb push_back

/*
__builtin_popcountll(x) : Number of 1-bit
__builtin_ctzll(x) : Number of trailing 0
*/

const long double PI = 3.1415926535897932384626433832795;
const int INF = 1000000000000000000;
const int MOD = 1000000007;
const int MOD2 = 1000000009;
const long double EPS = 1e-6;

using namespace std;

void add(int& a, int b) {
    if ((a += b) >= MOD) a -= MOD;
}

pii get(string& s) {
    int cur = 1, cur2 = 1, res = 0, res2 = 0;
    for (auto f : s) {
        add(res, cur * (f - 'a' + 1));
        add(res2, cur2 * (f - 'a' + 1));
        cur *= 47;
        cur2 *= 51;
    }
    return {res, res2};
}

int n;
string s[500005];
set<pii> se[500005];

signed main() {

    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);

    // freopen("cf.inp", "r", stdin);
    // freopen("cf.out", "w", stdout);

    cin >> n;
    for1(i,1,n) {
        cin >> s[i];
        if (i > 1 && s[i - 1][sz(s[i - 1]) - 1] != s[i][0]) {
            cout << "NO\n" << i;
            return 0;
        }
        pii x = get(s[i]);
        if (se[sz(s[i])].find(x) != se[sz(s[i])].end()) {
            cout << "NO\n" << i;
            return 0;
        }
        se[sz(s[i])].insert(x);
    }
    cout << "YES";

}

Comments

Please read the guidelines before commenting.



  • -1
    HN_CSP_FTRs  commented on Jan. 13, 2022, 1:45 p.m.

    map truy xuất mất log(n) mà bài này chưa bị quá thời gian à