QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#369808 | #8512. Harmonic Operations | willow | WA | 3ms | 10184kb | C++17 | 4.0kb | 2024-03-28 18:10:46 | 2024-03-28 18:10:47 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 4e5 + 5;
char s[maxn], t[maxn];
int n, k;
int Check(int x) {
for(int i = 1; i <= n; ++ i) {
if(s[i] != s[(i + x - 1) % n + 1]) {
return 0;
}
}
return 1;
}
int sh = -1, g;
struct Arr {
int del, rev, who, cnt[maxn];
void Init(int _who) {
del = rev = 0;
who = _who;
fill(cnt, cnt + maxn, 0);
}
void Reverse() {
rev ^= 1;
who ^= 1;
}
void Left(int w) {
if(rev) {
del += w;
}
else {
del -= w;
}
}
void Right(int w) {
if(rev) {
del -= w;
}
else {
del += w;
}
}
void Add(int x) {
// cerr << "Add delta = " << del << " rev = " << rev << " who = " << who << " x = " << x << endl;
if(rev) {
++ cnt[((-(del + x)) % g + g) % g];
}
else {
++ cnt[((x - del) % g + g) % g];
}
}
int Query() {
int tar = 0;
if(who) {
if(sh == -1)
return 0;
tar = sh;
}
// cerr << "Query tar = " << tar << endl;
if(rev) {
// cerr << "pos = " << (((-(del + tar)) % g + g) % g) << " " << cnt[((-(del + tar)) % g + g) % g] << endl;
return cnt[((-(del + tar)) % g + g) % g];
}
else {
// cerr << "pos = " << (((tar - del) % g + g) % g) << " " << cnt[((tar - del) % g + g) % g] << endl;
return cnt[((tar - del) % g + g) % g];
}
}
}a[2];
char op[5];
int fail[maxn];
main() {
scanf("%s", s + 1);
n = strlen(s + 1);
g = n;
for(int i = 1; i * i <= n; ++ i) {
if(n % i == 0) {
if(Check(i)) {
g = min(g, i);
}
if(Check(n / i)) {
g = min(g, n / i);
}
}
}
for(int i = 1; i <= n; ++ i) {
t[i + n] = t[i] = s[n - i + 1];
}
fail[1] = 0;
int p = 0;
for(int i = 2; i <= n; ++ i) {
while(p && s[i] != s[p + 1])
p = fail[p];
if(s[i] == s[p + 1])
++ p;
fail[i] = p;
}
int now = 0;
for(int i = 1; i <= 2 * n; ++ i) {
while(now && t[i] != s[now + 1])
now = fail[now];
if(t[i] == s[now + 1])
++ now;
// cerr << i << " " << now << endl;
if(now == n) {
sh = i - n;
break;
}
}
sh = ((g - sh) % g + g) % g;
// cerr << n << " " << g << " " << sh << endl;
a[0].Init(0), a[1].Init(1);
scanf("%lld", &k);
long long ans = 0;
for(int i = 1, w; i <= k; ++ i) {
scanf("%s", op + 1);
if(op[1] == 'I') { // reverse
a[0].Reverse();
a[1].Reverse();
if(a[0].who)
a[0].Add(0);
else
a[1].Add(0);
}
else {
scanf("%lld", &w);
if(op[1] == 'L') { // to_left w
a[0].Left(w);
a[1].Left(w);
if(!a[0].who)
a[0].Add(g - w);
else
a[1].Add(g - w);
}
else { // to_right w
a[0].Right(w);
a[1].Right(w);
if(!a[0].who)
a[0].Add(w);
else
a[1].Add(w);
}
}
// cerr << "a[0]: rev = " << a[0].rev << " delta = " << a[0].del << " who = " << a[0].who << ": ";
// for(int i = 0; i < g; ++ i)
// cerr << a[0].cnt[i] << " ";
// cerr << endl;
// cerr << "a[1]: rev = " << a[1].rev << " delta = " << a[1].del << " who = " << a[1].who << ": ";
// for(int i = 0; i < g; ++ i)
// cerr << a[1].cnt[i] << " ";
// cerr << endl;
ans += a[0].Query();
ans += a[1].Query();
// cerr << "? Right = " << i << " ans = " << ans << endl;
}
printf("%lld\n", ans);
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 9968kb
input:
pda 2 R 2 L 2
output:
1
result:
ok 1 number(s): "1"
Test #2:
score: 0
Accepted
time: 0ms
memory: 10144kb
input:
aaa 4 R 1 I I R 1
output:
10
result:
ok 1 number(s): "10"
Test #3:
score: -100
Wrong Answer
time: 3ms
memory: 10184kb
input:
caso 6 L 1 I I R 1 I I
output:
7
result:
wrong answer 1st numbers differ - expected: '4', found: '7'