QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#187899#7506. StrCartesianbulijiojiodibuliduo#AC ✓784ms225460kbC++176.1kb2023-09-25 05:12:592023-09-25 05:12:59

Judging History

你现在查看的是最新测评结果

  • [2023-09-25 05:12:59]
  • 评测
  • 测评结果:AC
  • 用时:784ms
  • 内存:225460kb
  • [2023-09-25 05:12:59]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;
#define rep(i,a,n) for (int i=a;i<n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define pb push_back
#define eb emplace_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
typedef vector<int> VI;
typedef basic_string<int> BI;
typedef long long ll;
typedef pair<int,int> PII;
typedef double db;
mt19937_64 mrand(2); 
const ll mod=1000000007;
int rnd(int x) { return mrand() % x;}
ll powmod(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;}
ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;}
// head

const int M=2010,N=50100,L=1010000;
string S;
vector<pair<string,int>> str[M],sb;
VI len;
PII va[N],vb[N];
int n,m,pid[L];
int ts[L*2];
string a[N],b[N];
char s[L];

int q;
ll l[M],r[M],md[M];

struct SuffixArray {

static const int N=2000100;
static const int LOGN=22;

int n;
int sa[N],rk[N],ht[N],so[N];
bool t[N<<1];
int hv[LOGN][N];

bool islms(const int i,const bool *t) {
	return i>0&&t[i]&&!t[i - 1];
}

template<class T>
inline void sort(T s,int *sa,const int len,const int sz,const int sigma,
					bool *t,int *b,int *cb,int *p) {
	memset(b,0,sizeof(int)*sigma);
	memset(sa,-1,sizeof(int)*len);
	rep(i,0,len) b[(int)s[i]]++;
	cb[0]=b[0];
	rep(i,1,sigma) cb[i]=cb[i-1]+b[i];
	per(i,0,sz) sa[--cb[(int)s[p[i]]]]=p[i];
	rep(i,1,sigma) cb[i]=cb[i-1]+b[i-1];
	rep(i,0,len) if (sa[i]>0&&!t[sa[i]-1]) sa[cb[(int)s[sa[i]-1]]++]=sa[i]-1;
	cb[0]=b[0];
	rep(i,1,sigma) cb[i]=cb[i-1]+b[i];
	per(i,0,len) if (sa[i]>0&&t[sa[i]-1]) sa[--cb[(int)s[sa[i]-1]]]=sa[i]-1;
}

template<class T>
inline void sais(T s,int *sa,const int len,bool *t,int *b,int *b1,
				const int sigma) {
	int p=-1,*cb=b+sigma;
	t[len-1]=1;
	per(i,0,len-1) t[i]=s[i]<s[i+1]||(s[i]==s[i+1]&&t[i+1]);
	int sz=0,cnt=0;
	rep(i,1,len) if (t[i]&&!t[i-1]) b1[sz++]=i;
	sort(s,sa,len,sz,sigma,t,b,cb,b1);
	sz=0;
	rep(i,0,len) if (islms(sa[i],t)) sa[sz++]=sa[i];
	rep(i,sz,len) sa[i]=-1;
	rep(i,0,sz) {
		int x=sa[i];
		rep(j,0,len) {
			if (p==-1||s[x+j]!=s[p+j]||t[x+j]!=t[p+j]) {
				cnt++; p=x;
				break;
			} else if (j>0&&(islms(x+j,t)||islms(p+j,t))) {
				break;
			}
		}
		sa[sz+(x>>=1)]=cnt-1;
	}
	for (int i=len-1,j=len-1;i>=sz;i--) if (sa[i]>=0) sa[j--]=sa[i];
	int *s1=sa+len-sz,*b2=b1+sz;
	if (cnt<sz) sais(s1,sa,sz,t+len,b,b1+sz,cnt);
	else rep(i,0,sz) sa[s1[i]]=i;
	rep(i,0,sz) b2[i]=b1[sa[i]];
	sort(s,sa,len,sz,sigma,t,b,cb,b2);
}

template<class T>
inline void getHeight(T s,int n) {
	rep(i,1,n+1) {
		//printf("sa %d ",sa[i]); puts("");
		rk[sa[i]]=i;
	}
	int j=0,k=0;
	for (int i=0;i<n;ht[rk[i++]]=k)
		for (k?k--:0,j=sa[rk[i]-1];s[i+k]==s[j+k];k++);
}

template<class T>
inline void init(T s,const int len,const int sigma) {
	sais(s,sa,len,t,rk,ht,sigma);
}

inline void solve(int *s,int len,int sigma=28) { //    0-based, s[len]=0
	s[len]=0;
	n=len;
	init(s,len+1,28);
	rep(i,0,len) so[i]=s[i];
	getHeight(s,len);
	rk[len]=0;
	rep(i,1,len+1) hv[0][i]=ht[i];
	rep(j,1,LOGN) for (int i=1;i+(1<<j)-1<=len;i++)
		hv[j][i]=min(hv[j-1][i],hv[j-1][i+(1<<(j-1))]);
}
int lcp(int p,int q) {    // 1-based
	if (q>n) return 0;
	if (p==q) return n-p+1;
	p=rk[p-1]; q=rk[q-1];
	if (p>q) swap(p,q);
	int w=31-__builtin_clz(q-p);
	return min(hv[w][p+1],hv[w][q-(1<<w)+1]);
}

int cmp(int p,int q,int len) { // 1-based p<=q
	int d=lcp(p,q);
	if (d>=len) return 0;
	assert(so[p+d-1]!=so[q+d-1]);
	return so[p+d-1]<so[q+d-1]?-1:1;
}

}sa;


int cmp(PII a,PII b) { // a<=b
	static PII ta[2],tb[2];
	ta[0]=va[a.fi]; ta[1]=vb[a.se];
	tb[0]=va[b.fi]; tb[1]=vb[b.se];
	int p1=0,p2=0;
	while (p1<2&&p2<2) {
		int l1=ta[p1].se-ta[p1].fi+1;
		int l2=tb[p2].se-tb[p2].fi+1;
		if (l1==l2) {
			int w=sa.cmp(ta[p1].fi,tb[p2].fi,l1);
			if (w!=0) return w;
			p1++; p2++;
		} else if (l1<l2) {
			int w=sa.cmp(ta[p1].fi,tb[p2].fi,l1);
			if (w!=0) return w;
			p1++;
			tb[p2].fi+=l1;
		} else {
			int w=sa.cmp(ta[p1].fi,tb[p2].fi,l2);
			if (w!=0) return w;
			p2++;
			ta[p1].fi+=l2;
		}
	}
	if (p1==2&&p2==2) return 0;
	if (p1==2) return -1;
	else return 1;
}


PII getpair(int lm,ll x) {	
	int p1=x/m,p2=x%m;
	return mp(str[lm][p1].se,sb[p2].se);
}
int main() {
	scanf("%d%d",&n,&m);
	rep(i,0,n) {
		scanf("%s",s);
		a[i]=s;
		va[i]=mp(SZ(S)+1,SZ(S)+SZ(a[i]));
		S+=a[i];
		len.pb(SZ(a[i]));

	}
	sort(all(len));
	len.erase(unique(all(len)),len.end());
	int lm=SZ(len);
	rep(i,0,lm) pid[len[i]]=i;
	rep(i,0,n) str[pid[SZ(a[i])]].pb(mp(a[i],i));
	rep(i,0,lm) sort(all(str[i]));
	rep(i,0,m) {
		scanf("%s",s);
		b[i]=s;
		vb[i]=mp(SZ(S)+1,SZ(S)+SZ(b[i]));
		S+=b[i];
		sb.pb(mp(b[i],i));
	}
	sort(all(sb));
	rep(i,0,SZ(S)) ts[i]=S[i]-'a'+1;
	sa.solve(ts,SZ(S));
	/*rep(i1,0,n*m) rep(i2,0,n*m) {
		string s1=a[i1/m]+b[i1%m];
		string s2=a[i2/m]+b[i2%m];
		int d=s1==s2?0:((s1<s2)?-1:1);
		printf("%d %d %d %d\n",i1/m,i1%m,i2/m,i2%m);
		assert(d==cmp(mp(i1/m,i1%m),mp(i2/m,i2%m)));
	}
	return 0;*/
	scanf("%d",&q);
	rep(i,0,q) {
		ll rk;
		scanf("%lld",&rk);
		rep(i,0,lm) l[i]=0,r[i]=(ll)SZ(str[i])*m-1;
		while (1) {
			ll val=0;
			rep(i,0,lm) if (l[i]<=r[i]) val+=r[i]-l[i]+1;
			ll pivot=mrand()%val;
			PII mid(-1,-1);
			rep(i,0,lm) if (l[i]<=r[i]) {
				ll rs=r[i]-l[i]+1;
				if (pivot>=rs) pivot-=rs;
				else {
					mid=getpair(i,l[i]+pivot);
					break;
				}
			}
			assert(mid!=mp(-1,-1));
			int ls=0;
			rep(i,0,lm) if (l[i]<=r[i]) {
				ll L=l[i]-1,R=r[i]+1;
				while (L+1<R) {
					ll M=(L+R)/2;
					PII w=getpair(i,M);
					if (cmp(w,mid)>=0) R=M; else L=M;
				}
				md[i]=R;
				ls+=md[i]-l[i];
			}
			if (ls<rk) {
				rk-=ls;
				rep(i,0,lm) if (l[i]<=r[i]) l[i]=md[i];
			} else {
				rep(i,0,lm) if (l[i]<=r[i]) r[i]=md[i]-1;
			}
			PII pr(-1,-1);
			rep(i,0,lm) if (l[i]<=r[i]) { pr=getpair(i,l[i]); break; }
			bool as=1;
			rep(i,0,lm) {
				if (l[i]<r[i]) {
					as=0;
				}
				if (l[i]==r[i]) {
					as&=cmp(pr,getpair(i,l[i]))==0;
				}
				if (!as) break;
			}
			if (as) {
				printf("%d %d\n",pr.fi+1,pr.se+1);
				break;
			}
		}
	}
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 30520kb

input:

2 3
a
ab
a
aa
ba
2
3
4

output:

1 3
1 3

result:

ok Correct Answer

Test #2:

score: 0
Accepted
time: 8ms
memory: 42832kb

input:

27 28
ptdtljwzjbyctrjetnossaiogcfce
kmuwnsxjenpjybcwrecoqddbltzvnkzzrxvne
lsbqkpbgihqahpotzrnggqrompohf
ixonxrpixmurqgdtedhavpawmddfw
ruhqnxhvucv
fvarlrf
uyxxjzjf
tlzrxaykw
kldatopsv
fywmszxti
yraazxvbfebdhekphdrqhvpmferaghezebm
dovtqdvdd
waohxsasgl
biakvxhqx
aeyzaqum
repmbqtrvwd
dvqojkduc
sjxapwnyz...

output:

23 1
2 27
4 11
2 11
21 9
11 13
7 6
27 1
5 26
17 3
21 23
1 10
6 25
13 5
17 20
22 22
1 5
23 21
10 23
3 18
10 14
16 14
18 5
22 26
14 6
12 22
19 17
14 17
23 28
10 20
26 15
27 6
19 28
16 27
21 25
16 1
6 8
10 20
12 19
1 10
17 10
14 14
18 19
15 27
10 9
9 27
4 19
8 1
17 7
21 1
17 26
26 13
22 13
27 7
2 28
15...

result:

ok Correct Answer

Test #3:

score: 0
Accepted
time: 646ms
memory: 221084kb

input:

21673 22789
osjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwnyzosjxapwnyaosjxapwn...

output:

11748 7566
13026 16461
9916 11268
12937 6138
20884 12212
310 19018
14665 17118
7698 15274
18832 22061
7551 10949
12970 7192
1098 9710
19691 12422
5351 11239
14696 2061
2889 12319
8909 10800
6739 1691
19504 5705
599 16293
2991 6114
20067 16922
11544 17617
7562 4195
7364 20824
19750 22736
15334 22205
...

result:

ok Correct Answer

Test #4:

score: 0
Accepted
time: 784ms
memory: 222920kb

input:

42793 43577
osjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjyosjxosjy...

output:

478 23079
19578 5700
32327 37245
23174 34890
8775 43551
33235 7846
40699 41435
24607 32335
10681 5081
40664 38308
4711 11678
31506 13791
13592 7796
9099 15409
35 36153
12575 30943
1418 37944
18258 11644
4315 19248
29431 9382
9938 30089
16317 36775
25241 10476
27724 29981
33560 26458
311 26256
24402 ...

result:

ok Correct Answer

Test #5:

score: 0
Accepted
time: 514ms
memory: 222220kb

input:

21455 22492
osjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaqosjxaposjxaq...

output:

1153 18391
12397 8491
9381 14158
8926 9710
19054 21286
18999 489
2337 3914
20214 18000
4041 11022
3232 2622
632 12989
4831 1494
7204 17465
7959 13995
4214 14108
14052 20985
1761 21505
3079 11058
16940 15409
4088 14365
14018 20169
19551 9309
16472 21758
13613 21356
6865 9708
11938 3822
14038 1038
370...

result:

ok Correct Answer

Test #6:

score: 0
Accepted
time: 546ms
memory: 225460kb

input:

33056 33045
yiyrokayfnzhamwfzuadvuyndjspptokvoqlgrgqytwceezsczyljaxyicjepttyvllsbqcfmsseskryayuezexpiypfgswfvzexhocjyzpxuazgrdjngsovhcrkdoirnxfxwjypnamxblzrqbgoomgltuoxkgtdnjsanhszewdkbediutypzjcasccewjnxakdeinqczivdzoecriaabjfuipztnvlapnqrsnqoigciixjrpgjxmtacshfrkeflmxjzmuwmkgpcsgqcgtlyqgugtdyoasns...

output:

13792 9065
8306 26578
7635 30555
18164 11919
11011 8423
25829 23538
30170 28490
19726 650
19941 2741
4050 1943
25996 22793
29283 32610
4160 4087
4759 31867
6046 6176
23450 14827
21825 16147
7681 18029
4173 15086
6943 7840
2462 27663
6954 18606
17654 23682
14061 2338
30610 1881
8076 25979
18709 32844...

result:

ok Correct Answer