QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#46043 | #1942. Another Substring Query Problem | vexentox | AC ✓ | 122ms | 55644kb | C++14 | 3.2kb | 2022-08-24 22:35:13 | 2022-08-24 22:35:14 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
constexpr int N=2e5+5;
constexpr int M=26;
int n,q;
string s;
vector<int> e[N],vec;
int tree_pos[N],k[N],len[N];
int dfn[N],dfn0,fi[N],lst[N];
int ans[N];
struct Trie{// AC机 P3808 P3796 P3121 P5357
int tot;
int nex[N][M];
int fail[N];
int cnt[N];
void insert(string s,int ind){
int len=s.size();
int u=0;
for(int i=0;i<len;i++){
int w=s[i]-'a';
if(!nex[u][w]) nex[u][w]=++tot;
u=nex[u][w];
}
cnt[u]++;
tree_pos[ind]=u;
}
void buildfail(){
queue<int> q;
for(int i=0;i<M;i++) if(nex[0][i]) q.push(nex[0][i]);
while(!q.empty()){
int u=q.front();
q.pop();
for(int i=0;i<M;i++){
if(nex[u][i]){
int v=nex[u][i];
fail[v]=nex[fail[u]][i];
q.push(nex[u][i]);
}
else nex[u][i]=nex[fail[u]][i];
}
}
for(int i=1;i<=tot;i++) e[fail[i]].push_back(i);
}
}tr;
struct BIT{//树状数组
int bit[N];
void add(int x,int k){
for(int i=x;i<=dfn0;i+=(i&-i)) bit[i]+=k;
}
int sum(int x){
int res=0;
for(int i=x;i;i-=(i&-i)) res+=bit[i];
return res;
}
}bit;
int main(){
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
cin>>s;
n=s.size();
cin>>q;
for(int i=1;i<=q;i++){
string t;
cin>>t>>k[i];
len[i]=t.size();
tr.insert(t,i);
}
tr.buildfail();
int now=0;
vec.push_back(now);
for(auto c:s){
now=tr.nex[now][c-'a'];
vec.push_back(now);
}
function<void(int)> dfs=[&](int x){
dfn[x]=++dfn0;
fi[x]=dfn0;
for(auto v:e[x]) dfs(v);
lst[x]=dfn0;
};
dfs(0);//预处理自动机fail树
function<void(vector<int>&,int,int)> solve=[&](vector<int> &qry,int l,int r){//整体二分 P3332 P1527 P3527 P3157
if(qry.size()==0) return;
if(l==r){
bit.add(dfn[vec[l]],1);
for(auto x:qry){
int sum=bit.sum(lst[tree_pos[x]])-bit.sum(fi[tree_pos[x]]-1);
if(k[x]==sum) ans[x]=l;
else ans[x]=-1;
}
bit.add(dfn[vec[l]],-1);
return;
}
int mid=l+r>>1;
vector<int> qry1,qry2;
for(int i=l;i<=mid;i++) bit.add(dfn[vec[i]],1);
for(auto x:qry){
int sum=bit.sum(lst[tree_pos[x]])-bit.sum(fi[tree_pos[x]]-1);
if(k[x]<=sum) qry1.push_back(x);
else{
k[x]-=sum;
qry2.push_back(x);
}
}
for(int i=l;i<=mid;i++) bit.add(dfn[vec[i]],-1);
solve(qry1,l,mid);
solve(qry2,mid+1,r);
};
vector<int> qry;
for(int i=1;i<=q;i++) qry.push_back(i);
solve(qry,1,n);
for(int i=1;i<=q;i++){
if(ans[i]==-1) cout<<ans[i]<<"\n";
else cout<<ans[i]-len[i]+1<<"\n";
}
return true&&false;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 10ms
memory: 17728kb
input:
hhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhh...
output:
50997 112451 97534 13355 150562 5772 146250 10904 4343 194777 71395 96501 6809 18571 166901 114789 123587 176339 29660 82650 63509 80101 47907 66143 199971 108032 140518 49221 46212 28127 61461 149025 142818 33646 17132 196161 36671 28656 16860 141951 139429 72866 162807 21343 194803 16263 118701 20...
result:
ok 631 lines
Test #2:
score: 0
Accepted
time: 97ms
memory: 28440kb
input:
gnzlzbsarybaxilacuopvdkhjfiynrgfdhsrecyxedcrriztcxboczhfaseqglcwshaewaepasxqpaxxbroxndbxqabrvtrrshrnnlyfodtxyltjkbghfkjcqrqsuwdsgypcaroaojqbfrjqbdrimglxcqtfmqaeogqunaegfpfzcmyjwfgbwaigwitcyspiedbisjuedeyseneuzjdteifzvdfzhenapmmreqpvwsmtrxowjnmmrthzhojgxsoppfgfgohoghuwvqclzsmeydusfndrdyqokpqwupixyoxr...
output:
-1 -1 174057 -1 141370 -1 75407 -1 112842 -1 -1 4510 100584 12223 35161 28768 -1 5258 151196 -1 25061 87769 -1 135272 -1 -1 -1 159851 -1 -1 121971 108762 60263 143662 -1 5200 65515 -1 -1 -1 -1 157064 65499 -1 84122 173745 171855 -1 193921 -1 121426 162134 137032 126244 99210 155648 95498 151755 1105...
result:
ok 54751 lines
Test #3:
score: 0
Accepted
time: 122ms
memory: 30964kb
input:
gnzlzbsarybaxilacuopvdkhjfiynrgfdhsrecyxedcrriztcxboczhfaseqglcwshaewaepasxqpaxxbroxndbxqabrvtrrshrnnlyfodtxyltjkbghfkjcqrqsuwdsgypcaroaojqbfrjqbdrimglxcqtfmqaeogqunaegfpfzcmyjwfgbwaigwitcyspiedbisjuedeyseneuzjdteifzvdfzhenapmmreqpvwsmtrxowjnmmrthzhojgxsoppfgfgohoghuwvqclzsmeydusfndrdyqokpqwupixyoxr...
output:
-1 -1 -1 66435 197872 -1 83779 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 91584 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 21808 -1 -1 -1 -1 -1 15575 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 196132 -...
result:
ok 200000 lines
Test #4:
score: 0
Accepted
time: 113ms
memory: 34908kb
input:
qzpazeynjuulxowssvbghqhdtywsnzgwmpospvxyeoxsrwxccwucqgsyllrkburhfrivzqcpnevmixivrtbxexhrjauxayxmrdluzkvxnfvocgoceavurwlxzdfepcxzcdaoppzfhqpyhmjlhbfsntaewlfjqgefvqttczahgzgucgoohnjsdvyaxltixdvbgzoewptjuosymanxeiewbvnibhjbkhoebygmtmwldtvyijasjquxvtwkhlngbcremdsprdetxkggogyjbgsuduzmutcgblzoqopwspbzrurs...
output:
109498 154508 -1 -1 -1 -1 1028 36133 -1 -1 -1 -1 -1 191213 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 129366 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 196816 -1 -1 141983 -1 -1 137732 -1 11350 -1 -1 -1 25285 52189 -1 -1 -1 -1 80999 -1 -1 17773 5870 -1 -1 -1 -1 -1 128813 -1 1492 -1 -1 1513 -1 -1 -1 139760 199034 -1 -1 49122 ...
result:
ok 50034 lines
Test #5:
score: 0
Accepted
time: 2ms
memory: 16100kb
input:
dkfjahfwbaofboahsdlvkjnalsf 10 dkfjahfwbaofboahsdlvkjnalsfd 1 adkfjahfwbaofboahsdlvkjnalsf 1 kfj 2 kfj 1 a 3 a 1 a 2 a 4 w 2 aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa 1
output:
-1 -1 -1 2 15 5 10 24 -1 -1
result:
ok 10 lines
Test #6:
score: 0
Accepted
time: 3ms
memory: 19400kb
input:
alfalfa 5 alfa 1 alfa 2 alfa 3 alfalfa 1 alfalfa 2
output:
1 4 -1 1 -1
result:
ok 5 lines
Test #7:
score: 0
Accepted
time: 3ms
memory: 16444kb
input:
x 28 a 1 b 1 c 1 d 1 e 1 f 1 g 1 h 1 i 1 j 1 k 1 l 1 m 1 n 1 o 1 p 1 q 1 r 1 s 1 t 1 u 1 v 1 w 1 x 1 y 1 z 1 abcdefghijklmnopqrstuvwxyz 1 abcdefghijklmnopqrstuvwxyz 1
output:
-1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 1 -1 -1 -1 -1
result:
ok 28 lines
Test #8:
score: 0
Accepted
time: 76ms
memory: 19476kb
input:
aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa...
output:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 ...
result:
ok 200000 lines
Test #9:
score: 0
Accepted
time: 17ms
memory: 38316kb
input:
aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa...
output:
100001 -1
result:
ok 2 lines
Test #10:
score: 0
Accepted
time: 26ms
memory: 55224kb
input:
aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa...
output:
2
result:
ok single line: '2'
Test #11:
score: 0
Accepted
time: 34ms
memory: 55644kb
input:
aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa...
output:
1
result:
ok single line: '1'
Test #12:
score: 0
Accepted
time: 19ms
memory: 18876kb
input:
abbabaabbaababbabaababbaabbabaabbaababbaabbabaababbabaabbaababbabaababbaabbabaababbabaabbaababbaabbabaabbaababbabaababbaabbabaabbaababbaabbabaababbabaabbaababbaabbabaabbaababbabaababbaabbabaababbabaabbaababbabaababbaabbabaabbaababbaabbabaababbabaabbaababbabaababbaabbabaababbabaabbaababbaabbabaabbaab...
output:
1 1537 3073 5121 6145 7681 9217 10241 12289 13825 15361 17409 18433 20481 22017 23553 24577 26113 27649 29697 30721 32257 33793 34817 36865 38401 39937 40961 42497 44033 46081 47105 49153 50689 52225 54273 55297 56833 58369 59393 61441 62977 64513 66561 67585 69633 71169 72705 73729 75265 76801 7884...
result:
ok 195 lines