QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#280288 | #7780. Dark LaTeX vs. Light LaTeX | ucup-team1525# | TL | 742ms | 394652kb | C++20 | 1.6kb | 2023-12-09 14:56:49 | 2023-12-09 14:56:50 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
const int N=5e3;
const int mod=998244353,v=1919810;
int n,m;
char s[N+5],t[N+5];
int hs[N+5][N+5],ht[N+5][N+5];
unordered_map<int,int> mp[N+5];
ll vals[N+5][N+5];
ll ans;
void work(int l){
static int z[N+5];
memset(z,0,sizeof z);
for(int i=l+1,j=l,r=0;i<=m;i++){
int &k=z[i]=(r<=i?0:min(z[i-j+l],r-i));
while(i+k<=m&&t[l+k]==t[i+k]) k++;
if(i+k>r){
j=i; r=i+k;
}
if(z[i]>0)
ans+=vals[i-1][min(i-1,l+z[i])]-vals[i-1][l];
}
}
void solve(){
n=strlen(s+1); m=strlen(t+1);
for(int i=1;i<=max(n,m);i++) mp[i].clear();
for(int l=0;l<n;l++)
for(int i=1;i+l<=n;i++)
mp[l+1][hs[i][i+l]]++;
for(int i=1;i<=m;i++)
for(int j=i;j<=m;j++)
vals[j][i]=mp[j-i+1][ht[i][j]];
for(int j=1;j<=m;j++)
for(int i=1;i<=j;i++)
vals[j][i]+=vals[j][i-1];
for(int i=1;i<=m;i++)work(i);
}
int main(){
scanf("%s",s+1);
scanf("%s",t+1);
n=strlen(s+1); m=strlen(t+1);
for(int i=1;i<=n;i++){
int val=0;
for(int j=i;j<=n;j++){
val=(1ll*v*val+s[j]-'a')%mod;
hs[i][j]=val;
}
}
for(int i=1;i<=m;i++){
int val=0;
for(int j=i;j<=m;j++){
val=(1ll*v*val+t[j]-'a')%mod;
ht[i][j]=val;
}
}
solve();
swap(s,t);
swap(hs,ht);
solve();
// printf("%lld\n",ans);
for(int i=1;i<=m;i++) ans+=vals[i][i];
printf("%lld\n",ans);
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 20ms
memory: 200568kb
input:
abab ab
output:
8
result:
ok 1 number(s): "8"
Test #2:
score: 0
Accepted
time: 7ms
memory: 201512kb
input:
abab abaaab
output:
29
result:
ok 1 number(s): "29"
Test #3:
score: 0
Accepted
time: 15ms
memory: 203264kb
input:
abcd abcde
output:
10
result:
ok 1 number(s): "10"
Test #4:
score: 0
Accepted
time: 15ms
memory: 201476kb
input:
aaba ba
output:
6
result:
ok 1 number(s): "6"
Test #5:
score: 0
Accepted
time: 24ms
memory: 202172kb
input:
babababaaabbaabababbbaabbbababbaaaaa aaaabbaababbab
output:
1161
result:
ok 1 number(s): "1161"
Test #6:
score: 0
Accepted
time: 742ms
memory: 394652kb
input:
aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa...
output:
78156256250000
result:
ok 1 number(s): "78156256250000"
Test #7:
score: 0
Accepted
time: 96ms
memory: 253268kb
input:
gzgzggzgzggzggzgzggzgzggzggzgzggzggzgzggzgzggzggzgzggzggzgzggzgzggzggzgzggzgzggzggzgzggzggzgzggzgzggzggzgzggzgzggzggzgzggzggzgzggzgzggzggzgzggzggzgzggzgzggzggzgzggzgzggzggzgzggzggzgzggzgzggzggzgzggzggzgzggzgzggzggzgzggzgzggzggzgzggzggzgzggzgzggzggzgzggzgzggzggzgzggzggzgzggzgzggzggzgzggzggzgzggzgzggz...
output:
60716448
result:
ok 1 number(s): "60716448"
Test #8:
score: 0
Accepted
time: 114ms
memory: 254316kb
input:
mlmllmllmlmllmlmllmllmlmllmlmllmllmlmllmllmlmllmlmllmllmlmllmlmllmllmlmllmllmlmllmlmllmllmlmllmllmlmllmlmllmllmlmllmlmllmllmlmllmllmlmllmlmllmllmlmllmllmlmllmlmllmllmlmllmlmllmllmlmllmllmlmllmlmllmllmlmllmlmllmllmlmllmllmlmllmlmllmllmlmllmllmlmllmlmllmllmlmllmlmllmllmlmllmllmlmllmlmllmllmlmllmlmllml...
output:
60679828
result:
ok 1 number(s): "60679828"
Test #9:
score: -100
Time Limit Exceeded
input:
vbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbbvbvbbvbvbbvbbvbvbbvbbvbvbbvbvbbvb...