This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
#ifdef LOCAL
#include <debug.h>
#endif
#define st first
#define nd second
#define pb push_back
#define pf push_front
#define _pb pop_back
#define _pf pop_front
#define lb lower_bound
#define ub upper_bound
#define mtp make_tuple
#define all(x) (x).begin(), (x).end()
#define sz(x) (int)(x).size()
typedef long long ll; typedef unsigned long long ull;
typedef double db; typedef long double ldb;
typedef pair<int, int> pi; typedef pair<ll, ll> pll;
typedef vector<int> vi; typedef vector<ll> vll; typedef vector<pi> vpi; typedef vector<pll> vpll;
typedef string str;
template<typename T> T gcd(T a, T b) { return (b == 0? a : gcd(b, a % b)); }
template<typename T> T lcm(T a, T b) { return a / gcd(a, b) * b; }
#define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
#define FOS(i, r, l) for (int (i) = (r); (i) >= (l); --(i))
#define EACH(i, x) for (auto &(i) : (x))
#define WHILE while
#define file "TEST"
mt19937 rd(chrono::steady_clock::now().time_since_epoch().count());
ll rand(ll l, ll r) { return uniform_int_distribution<ll>(l, r)(rd); }
/*
Tran The Bao
CTL - Da Lat
Cay ngay cay dem nhung deo duoc cong nhan
*/
#pragma GCC target ("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
const int oo = 1e9;
const int T = 4e5 + 5;
struct MinSegtree {
int tr[T];
void upd(int i, int l, int r, int q, int v) {
if (r < q || l > q) return;
if (l == r) {
tr[i] = v;
return;
}
int mid = l + (r - l) / 2;
upd(i * 2, l, mid, q, v);
upd(i * 2 + 1, mid + 1, r, q, v);
tr[i] = min(tr[i * 2], tr[i * 2 + 1]);
}
int get(int i, int l, int r, int ql, int qr) {
if (r < ql || l > qr) return oo;
if (ql <= l && r <= qr) return tr[i];
int mid = l + (r - l) / 2;
return min(get(i * 2, l, mid, ql, qr), get(i * 2 + 1, mid + 1, r, ql, qr));
}
};
struct Mergetree {
vi tr[T];
void add(int i, int l, int r, int q, int v) {
if (r < q || l > q) return;
tr[i].pb(v);
if (l == r) return;
int mid = l + (r - l) / 2;
add(i * 2, l, mid, q, v);
add(i * 2 + 1, mid + 1, r, q, v);
}
void dfs(int i, int l, int r) {
sort(all(tr[i]));
if (l == r) return;
int mid = l + (r - l) / 2;
dfs(i * 2, l, mid);
dfs(i * 2 + 1, mid + 1, r);
}
int get(int i, int l, int r, int ql, int qr, int v) {
if (r < ql || l > qr) return 0;
if (ql <= l && r <= qr) {
if (tr[i].empty() || tr[i].back() < v) return 0;
int tmp = lb(all(tr[i]), v) - tr[i].begin();
return sz(tr[i]) - tmp;
}
int mid = l + (r - l) / 2;
return get(i * 2, l, mid, ql, qr, v) + get(i * 2 + 1, mid + 1, r, ql, qr, v);
}
};
const pi M = {1e9 + 7, 1e9 + 9};
const pi base = {37, 127};
const int N = 1e5 + 5;
int n;
pi P[N];
str s;
pi q[N];
namespace subtask12 {
int p[N];
str a1[N];
set<pi> a2[N];
int getp(int i) { return i == p[i]? i : p[i] = getp(p[i]); }
int join(int x, int y) {
x = getp(x);
y = getp(y);
if (sz(a2[x]) < sz(a2[y])) swap(a2[x], a2[y]);
EACH(i, a2[y]) a2[x].insert(i);
int m1 = sz(a1[x]);
int m2 = sz(a1[y]);
vpi h(m1 + m2 + 2), h1(m1 + m2 + 2);
a1[x] += a1[y];
auto get = [&](int l, int r) -> pi {
pi rs;
rs.st = (h[r].st - 1LL * (l == 0? 0 : h[l - 1].st) * P[r - l + 1].st % M.st + M.st) % M.st;
rs.nd = (h[r].nd - 1LL * (l == 0? 0 : h[l - 1].nd) * P[r - l + 1].nd % M.nd + M.nd) % M.nd;
return rs;
};
auto get1 = [&](int l, int r) -> pi {
pi rs;
rs.st = (h1[l].st - 1LL * h1[r + 1].st * P[r - l + 1].st % M.st + M.st) % M.st;
rs.nd = (h1[l].nd - 1LL * h1[r + 1].nd * P[r - l + 1].nd % M.nd + M.nd) % M.nd;
return rs;
};
pi cur = {0, 0};
FOR(i, 0, m1 + m2 - 1) {
cur.st = (1LL * cur.st * base.st % M.st + a1[x][i] - '0' + 1) % M.st;
cur.nd = (1LL * cur.nd * base.nd % M.nd + a1[x][i] - '0' + 1) % M.nd;
h[i] = cur;
}
cur = {0, 0};
FOS(i, m1 + m2 - 1, 0) {
cur.st = (1LL * cur.st * base.st % M.st + a1[x][i] - '0' + 1) % M.st;
cur.nd = (1LL * cur.nd * base.nd % M.nd + a1[x][i] - '0' + 1) % M.nd;
h1[i] = cur;
}
FOR(i, 0, m1 - 1)
FOR(j, m1, m1 + m2 - 1) {
pi tmp = get(i, j);
if (tmp == get1(i, j)) a2[x].insert(tmp);
}
p[y] = x;
return sz(a2[x]);
}
void solve() {
FOR(i, 1, n) {
p[i] = i;
a1[i].pb(s[i]);
a2[i].insert({s[i] - '0' + 1, s[i] - '0' + 1});
}
FOR(i, 1, n - 1) {
int a = q[i].st, b = q[i].nd;
cout << join(a, b) << '\n';
}
}
}
namespace subtask3 {
pi h[N], h1[N];
int sa[N], d[N], d1[N];
pi get(int l, int r) {
pi rs;
rs.st = (h[r].st - 1LL * h[l - 1].st * P[r - l + 1].st % M.st + M.st) % M.st;
rs.nd = (h[r].nd - 1LL * h[l - 1].nd * P[r - l + 1].nd % M.nd + M.nd) % M.nd;
return rs;
}
pi get1(int l, int r) {
pi rs;
rs.st = (h1[l].st - 1LL * h1[r + 1].st * P[r - l + 1].st % M.st + M.st) % M.st;
rs.nd = (h1[l].nd - 1LL * h1[r + 1].nd * P[r - l + 1].nd % M.nd + M.nd) % M.nd;
return rs;
}
int lcp(int a, int b) {
int l = 1, r = min(n - a + 1, n - b + 1), ans = 0;
WHILE(l <= r) {
int mid = l + (r - l) / 2;
if (get(a, a + mid - 1) == get(b, b + mid - 1)) {
ans = mid;
l = mid + 1;
}
else r = mid - 1;
}
return ans;
}
bool cmp(int a, int b) {
int ans = lcp(a, b);
if (ans == n - a + 1) return 1;
if (ans == n - b + 1) return 0;
return s[a + ans] < s[b + ans];
}
int LB(int a, int b) {
int l = 1, r = n, ans = -1;
WHILE(l <= r) {
int mid = l + (r - l) / 2;
int tmp = lcp(sa[mid], l);
if (tmp < b - a + 1) {
if (tmp == n - sa[mid] + 1) l = mid + 1;
else if (s[sa[mid] + tmp] < s[a + mid]) l = mid + 1;
else r = mid - 1;
}
else {
ans = mid;
r = mid - 1;
}
}
return ans;
}
int UB(int a, int b) {
int l = 1, r = n, ans = -1;
WHILE(l <= r) {
int mid = l + (r - l) / 2;
int tmp = lcp(sa[mid], l);
if (tmp < b - a + 1) {
if (tmp == n - sa[mid] + 1) l = mid + 1;
else if (s[sa[mid] + tmp] < s[a + mid]) l = mid + 1;
else r = mid - 1;
}
else {
ans = mid;
l = mid + 1;
}
}
return ans;
}
void solve() {
FOR(i, 1, n) {
sa[i] = i;
h[i].st = (1LL * h[i - 1].st * base.st % M.st + s[i] - '0' + 1) % M.st;
h[i].nd = (1LL * h[i - 1].nd * base.nd % M.nd + s[i] - '0' + 1) % M.nd;
}
FOS(i, n, 1) {
sa[i] = i;
h1[i].st = (1LL * h1[i + 1].st * base.st % M.st + s[i] - '0' + 1) % M.st;
h1[i].nd = (1LL * h1[i + 1].nd * base.nd % M.nd + s[i] - '0' + 1) % M.nd;
}
FOR(i, 1, n) {
int l = 1, r = min(i, n - i + 1), ans = 0;
WHILE(l <= r) {
int mid = l + (r - l) / 2;
if (get(i - mid + 1, i + mid - 1) == get1(i - mid + 1, i + mid - 1)) {
ans = mid;
l = mid + 1;
}
else r = mid - 1;
}
d[i] = ans;
}
FOR(i, 1, n - 1) {
int l = 1, r = min(i, n - i), ans = 0;
WHILE(l <= r) {
int mid = l + (r - l) / 2;
if (get(i - mid + 1, i + mid) == get1(i - mid + 1, i + mid)) {
ans = mid;
l = mid + 1;
}
else r = mid - 1;
}
d1[i] = ans;
}
sort(sa + 1, sa + 1 + n, cmp);
MinSegtree s;
Mergetree s1, s2;
FOR(i, 1, n) {
s1.add(1, 1, n, i, i + d[i] - 1);
if (d1[i]) s2.add(1, 1, n, i, i + d1[i]);
}
s1.dfs(1, 1, n);
s2.dfs(1, 1, n);
FOR(i, 1, n) s.upd(1, 1, n, i, sa[i]);
int l = 2;
ll rs = 1;
FOR(i, 2, n) {
WHILE(l <= i) {
int L = LB(l, i), R = UB(l, i);
if (L == -1 || s.get(1, 1, n, L, R) >= l) {
++l;
continue;
}
break;
}
if (l > i) continue;
int tmp = (l + i) / 2;
if (tmp) {
rs += s1.get(1, 1, n, 1, tmp, i);
rs += s2.get(1, 1, n, 1, tmp, i);
}
cout << rs << '\n';
}
}
}
signed main() {
ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);
// freopen(file".inp", "r", stdin);
// freopen(file".out", "w", stdout);
cin >> n >> s;
P[0] = {1, 1};
FOR(i, 1, n) {
P[i].st = 1LL * P[i - 1].st * base.st % M.st;
P[i].nd = 1LL * P[i - 1].nd * base.nd % M.nd;
}
s = "#" + s + "#";
FOR(i, 1, n - 1) cin >> q[i].st >> q[i].nd;
if (n <= 1000) {
subtask12::solve();
return 0;
}
subtask3::solve();
return 0;
}
/*
3
010
1 2
2 3
*/
Compilation message (stderr)
Main.cpp:37: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
37 | #pragma GCC optimization ("O3")
|
Main.cpp:38: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
38 | #pragma GCC optimization ("unroll-loops")
|
Main.cpp: In function 'int subtask12::join(int, int)':
Main.cpp:26:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
26 | #define EACH(i, x) for (auto &(i) : (x))
| ^
Main.cpp:105:3: note: in expansion of macro 'EACH'
105 | EACH(i, a2[y]) a2[x].insert(i);
| ^~~~
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:123:3: note: in expansion of macro 'FOR'
123 | FOR(i, 0, m1 + m2 - 1) {
| ^~~
Main.cpp:25:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
25 | #define FOS(i, r, l) for (int (i) = (r); (i) >= (l); --(i))
| ^
Main.cpp:129:3: note: in expansion of macro 'FOS'
129 | FOS(i, m1 + m2 - 1, 0) {
| ^~~
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:134:3: note: in expansion of macro 'FOR'
134 | FOR(i, 0, m1 - 1)
| ^~~
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'j' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:135:3: note: in expansion of macro 'FOR'
135 | FOR(j, m1, m1 + m2 - 1) {
| ^~~
Main.cpp: In function 'void subtask12::solve()':
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:143:3: note: in expansion of macro 'FOR'
143 | FOR(i, 1, n) {
| ^~~
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:148:3: note: in expansion of macro 'FOR'
148 | FOR(i, 1, n - 1) {
| ^~~
Main.cpp: In function 'void subtask3::solve()':
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:222:3: note: in expansion of macro 'FOR'
222 | FOR(i, 1, n) {
| ^~~
Main.cpp:25:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
25 | #define FOS(i, r, l) for (int (i) = (r); (i) >= (l); --(i))
| ^
Main.cpp:227:3: note: in expansion of macro 'FOS'
227 | FOS(i, n, 1) {
| ^~~
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:232:3: note: in expansion of macro 'FOR'
232 | FOR(i, 1, n) {
| ^~~
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:244:3: note: in expansion of macro 'FOR'
244 | FOR(i, 1, n - 1) {
| ^~~
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:259:3: note: in expansion of macro 'FOR'
259 | FOR(i, 1, n) {
| ^~~
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:265:3: note: in expansion of macro 'FOR'
265 | FOR(i, 1, n) s.upd(1, 1, n, i, sa[i]);
| ^~~
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:268:3: note: in expansion of macro 'FOR'
268 | FOR(i, 2, n) {
| ^~~
Main.cpp: In function 'int main()':
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:293:5: note: in expansion of macro 'FOR'
293 | FOR(i, 1, n) {
| ^~~
Main.cpp:24:31: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
24 | #define FOR(i, l, r) for (int (i) = (l); (i) <= (r); ++(i))
| ^
Main.cpp:298:5: note: in expansion of macro 'FOR'
298 | FOR(i, 1, n - 1) cin >> q[i].st >> q[i].nd;
| ^~~| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
