BZOJ3439: Kpm的MC密码

题解:很显然的 我们需要对于每个串按照后缀建树 我们需要hash一下处理 相同串的情况 所以说对于每个串我们需要这个串出现的下标 然后按照长度排序插入trie中 对于这个串中经过的sum标记的结点 将这个串的价值插入到可持久化线段树中 查询按照区间第K大查询即可(亲测bzoj可以关同步23333

#include <bits/stdc++.h>
const int MAXN=2e5+10;
const int NM=3e5+10;
using namespace std;
typedef struct node{
    string s;int len;
    vector<int>v;
    friend bool operator<(node aa,node bb){
	return aa.len<bb.len;
    }
}node;
node d[MAXN];
map<string,int>ma;
map<string,int>::iterator ite;
string s1;
typedef struct trie{
    int a[26];int sum,rt;
}trie;
trie t[NM];int n;
int num,root,num1;
int newnode(){
    int x=++num;
    for(int i=0;i<26;i++)t[x].a[i]=0;
    t[x].sum=0;
    return x;
}
typedef struct Node{
    int l,r;int sum;
}Node;
Node p[51*MAXN];int vis[MAXN];
void update(int &x,int y,int l,int r,int t){
    x=++num1;p[x]=p[y];p[x].sum++;
    if(l==r)return ;
    int mid=(l+r)>>1;
    if(t<=mid)update(p[x].l,p[y].l,l,mid,t);
    else update(p[x].r,p[y].r,mid+1,r,t);
}
int ans1,sum1;
void querty(int x,int l,int r,int k){
    if(l==r){ans1=l;return ;}
    int mid=(l+r)>>1;
    if(p[p[x].l].sum>=k)querty(p[x].l,l,mid,k);
    else querty(p[x].r,mid+1,r,k-p[p[x].l].sum);
}
void insert(int pos){
    int temp=root;int len=d[pos].len;
    for(int i=0;i<len;i++){
	int t1=d[pos].s[i]-'a';
	if(!t[temp].a[t1])t[temp].a[t1]=newnode();
	temp=t[temp].a[t1];
	if(t[temp].sum){
	    for(int j=0;j<d[pos].v.size();j++)update(t[temp].rt,t[temp].rt,1,n,d[pos].v[j]);
	}
    }
    t[temp].sum=1;
    for(int i=0;i<d[pos].v.size();i++)update(t[temp].rt,t[temp].rt,1,n,d[pos].v[i]),vis[d[pos].v[i]]=temp;
}
int main(){
    std::ios::sync_with_stdio(false);
    cin>>n;int cnt=0;root=newnode();
    for(int i=1;i<=n;i++){
	cin>>s1;reverse(s1.begin(),s1.end());
	ite=ma.find(s1);
	if(ite==ma.end())d[++cnt].s=s1,d[cnt].v.push_back(i),ma[s1]=cnt,d[cnt].len=s1.length();
	else d[ite->second].v.push_back(i);
    }
    sort(d+1,d+cnt+1);
    for(int i=1;i<=cnt;i++)insert(i);
    int k;
    for(int i=1;i<=n;i++){
	cin>>k;
	if(k>p[t[vis[i]].rt].sum)puts("-1");
	else{
	    querty(t[vis[i]].rt,1,n,k);
	    printf("%d\n",ans1);
	}
    }
    return 0;
}

 

3439: Kpm的MC密码

Time Limit: 15 Sec  Memory Limit: 256 MB
Submit: 1093  Solved: 504
[Submit][Status][Discuss]

Description


 背景

    想Kpm当年为了防止别人随便进入他的MC,给他的PC设了各种奇怪的密码和验证问题(不要问我他是怎么设的。。。),于是乎,他现在理所当然地忘记了密码,只能来解答那些神奇的身份验证问题了。。。

 描述

    Kpm当年设下的问题是这样的:

    现在定义这么一个概念,如果字符串s是字符串c的一个后缀,那么我们称c是s的一个kpm串。

    系统将随机生成n个由a…z组成的字符串,由1…n编号(s1,s2…,sn),然后将它们按序告诉你,接下来会给你n个数字,分别为k1…kn,对于每一个ki,要求你求出列出的n个字符串中所有是si的kpm串的字符串的编号中第ki小的数,如果不存在第ki小的数,则用-1代替。(比如说给出的字符串是cd,abcd,bcd,此时k1=2,那么”cd”的kpm串有”cd”,”abcd”,”bcd”,编号分别为1,2,3其中第2小的编号就是2)(PS:如果你能在相当快的时间里回答完所有n个ki的查询,那么你就可以成功帮kpm进入MC啦~~)

Input

 

    第一行一个整数 n 表示字符串的数目

    接下来第二行到n+1行总共n行,每行包括一个字符串,第i+1行的字符串表示编号为i的字符串

    接下来包括n行,每行包括一个整数ki,意义如上题所示

 

Output

 

    包括n行,第i行包括一个整数,表示所有是si的kpm串的字符串的编号中第ki小的数

 

Sample Input


3
cd
abcd
bcd
2
3
1

Sample Output

2
-1
2

样例解释

“cd”的kpm 串有”cd”,”abcd”,”bcd”,编号为1,2,3,第2小的编号是

2,”abcd”的kpm串只有一个,所以第3小的编号不存在,”bcd”的kpm

串有”abcd”,”bcd”,第1小的编号就是2。

数据范围与约定

设所有字符串的总长度为len


对于100%的数据,1<=n<=100000,0

posted @ 2018-06-23 23:27  wang9897  阅读(213)  评论(0编辑  收藏  举报