2021HDU多校10 - 7084 Pty loves string(KMPnext树+主席树+dfs序)

Posted Frozen_Guardian

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了2021HDU多校10 - 7084 Pty loves string(KMPnext树+主席树+dfs序)相关的知识,希望对你有一定的参考价值。

题目链接:点击查看

题目大意:给出一个长度为 n n n 的字符串 s s s,需要回答 q q q 次询问,每次询问给出一对 ( x , y ) (x,y) (x,y) ,意思是用 s s s 的前缀 s [ 1 : x ] s[1:x] s[1:x] 和后缀 s [ n − y + 1 : n ] s[n-y+1:n] s[ny+1:n] 拼起来后得到一个新的字符串 t t t,问 t t t s s s 中的出现次数

题目分析:考虑对于一对询问 ( x , y ) (x,y) (x,y) 得到的 t t t,存在 s [ l : r ] = t s[l:r]=t s[l:r]=t,设断点为 x ′ = l + x − 1 , y ′ = r − y + 1 x'=l+x-1,y'=r-y+1 x=l+x1,y=ry+1 不难看出 s [ l : r ] s[l:r] s[l:r] 可以拆成 s [ l : x ′ ] + s [ y ′ : r ] s[l:x']+s[y':r] s[l:x]+s[y:r],满足
s [ l : x ′ ] = s [ 1 : x ] s[l:x']=s[1:x] s[l:x]=s[1:x] s [ y ′ : r ] = s [ n − y + 1 : n ] s[y':r]=s[n-y+1:n] s[y:r]=s[ny+1:n]

如果对于 K M P KMP KMP 正着、倒着分别建出 n e x t next next 树后,上述公式的意义是:

  1. x ′ x' x 一定位于点 x x x 的子树中
  2. y ′ y' y 一定位于点 n − y + 1 n-y+1 ny+1 的子树中

所以问题转换为了,两棵子树中,有多少对 ( x ′ , y ′ ) (x',y') (x,y) ,满足 x ′ + 1 = y ′ x'+1=y' x+1=y

类比与子树求交问题,我们只需要在一棵树上沿着 d f s dfs dfs 序建立主席树,维护的是第二棵树上 d f s dfs dfs 序就可以快速求解了。

代码:

// Problem: Pty loves string
// Contest: Virtual Judge - HDU
// URL: https://vjudge.net/problem/HDU-7084
// Memory Limit: 524 MB
// Time Limit: 10000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

// #pragma GCC optimize(2)
// #pragma GCC optimize("Ofast","inline","-ffast-math")
// #pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
#include<bitset>
#include<list>
#include<unordered_map>
#define lowbit(x) (x&-x)
using namespace std;
typedef long long LL;
typedef unsigned long long ull;
template<typename T>
inline void read(T &x)
{
	T f=1;x=0;
	char ch=getchar();
	while(0==isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
	while(0!=isdigit(ch)) x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
	x*=f;
}
template<typename T>
inline void write(T x)
{
	if(x<0){x=~(x-1);putchar('-');}
    if(x>9)write(x/10);
    putchar(x%10+'0');
}
const int inf=0x3f3f3f3f;
const int N=1e6+100;
struct Node {
	int l,r,sum;
}tree[N*20];
char s[N];
int nt[N],L1[N],R1[N],L2[N],R2[N],dfn1,dfn2,rk[N],root[N],cnt;
vector<int>node1[N],node2[N];
void dfs1(int u,int fa) {
	L1[u]=++dfn1;
	rk[dfn1]=u;
	for(auto v:node1[u]) {
		if(v==fa) {
			continue;
		}
		dfs1(v,u);
	}
	R1[u]=dfn1;
}
void dfs2(int u,int fa) {
	L2[u]=++dfn2;
	for(auto v:node2[u]) {
		if(v==fa) {
			continue;
		}
		dfs2(v,u);
	}
	R2[u]=dfn2;
}
void update(int num,int &k,int l,int r)
{
	tree[cnt++]=tree[k];
	k=cnt-1;
	tree[k].sum++;
	if(l==r) return;
	int mid=(l+r)>>1;
	if(num<=mid) update(num,tree[k].l,l,mid);
	else update(num,tree[k].r,mid+1,r);
}
int query(int i,int j,int l,int r,int L,int R) {//[L,R]:cur [l,r]:target
	if(L>r||R<l) return 0;
	if(L>=l&&R<=r) return tree[j].sum-tree[i].sum;
	int mid=(L+R)>>1;
	return query(tree[i].l,tree[j].l,l,r,L,mid)+query(tree[i].r,tree[j].r,l,r,mid+1,R);
}
void get_next() {
	int n=strlen(s+1);
	nt[1]=0;
	for(int i=2,j=0;i<=n;i++) {
		while(j!=0&&s[j+1]!=s[i]) {
			j=nt[j];
		}
		if(s[j+1]==s[i]) {
			j++;
		}
		nt[i]=j;
	}
	for(int i=1;i<=n;i++) node1[nt[i]].push_back(i);
	nt[n]=n+1;
	for(int i=n-1,j=n+1;i>=1;i--) {
		while(j!=n+1&&s[j-1]!=s[i]) {
			j=nt[j];
		}
		if(s[j-1]==s[i]) {
			j--;
		}
		nt[i]=j;
	}
	for(int i=1;i<=n;i++) node2[nt[i]].push_back(i);
}
void init(int n) {
	for(int i=0;i<=n+5;i++) {
		node1[i].clear(),node2[i].clear();
	}
	root[0]=0;
	tree[0].l=tree[0].r=tree[0].sum=0;
	cnt=1;
	dfn1=dfn2=0;
}
int main()
{
#ifndef ONLINE_JUDGE
//	freopen("data.in.txt","r",stdin);
//	freopen("data.out.txt","w",stdout);
#endif
//	ios::sync_with_stdio(false);
	int w;
	cin>>w;
	while(w--) {
		int n,m;
		scanf("%d%d%s",&n,&m,s+1);
		init(n);
		get_next()2021HDU多校6 - 7028 Decomposition(构造)

2021HDU多校6 - 7029 Median(思维)

2021 HDU多校 Boring data structure problem 链表模拟

2021 HDU多校 Boring data structure problem 链表模拟

2021HDU多校第三场 1009 Rise in Price 思维+乱搞

2021HDU多校8 - 7059 Counting Stars(线段树)