题目大意:有n个字符串s1~sn,n个数k1~kn,求以si字符串为后缀的所有字符串中第ki个字符串的编号。看样例解释就很容易理解了,不做赘述。
学习了大佬的字符串哈希解法。。原地址。。
想法是把每一个后缀哈希,然后记录每个后缀的字符串编号。然后把每个后缀编个号,存储一下每个编号的哈希值,用链表存储和查找。
#include<iostream>
#include<string>
#include<cstring>
#include<stdlib.h>
#include<cstdio>
#include<stdio.h>
#include<set>
#include<map>
#include<deque>
#include<stack>
#include<vector>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<list>
#include<bitset>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxint = ~0U >> 1;
const int maxn = 300010;
const int bas = 233;
const int M = 100007;
vector<int>mp[maxn];
char str[maxn];
int h[M];
int head[M], nxt[M], cnt;
ull val[M];
int insert(ull hs)//插入哈希值,同时也是获取该哈希值的编号
{
int u = hs % M;
for (int i = head[u];~i;i = nxt[i])//因为几个哈希值膜完是同一个值,用链表存储
{
if (val[i] == hs)//如果已经有这个后缀
return i;
}
nxt[++cnt] = head[u];
val[cnt] = hs;
head[u] = cnt;
return cnt;
}
int main()
{
int n, m, len,hs;
scanf("%d", &n);
memset(head, -1, sizeof head);
cnt = 0;
for (int i = 1;i <= n;i++)
{
scanf("%s", str);
len = strlen(str);hs = 0;
for (int t = len - 1;t >= 0;t--)
{
hs = hs * bas + str[t];
mp[insert(hs)].push_back(i);//第i个字符串存在哈希值为hs的后缀,所以记录到相应后缀的vector里
}
h[i] = hs;
}
for (int i = 1;i <= n;i++)
{
scanf("%d", &m);int p = insert(h[i]);
if (mp[p].size() < m)printf("-1\n");
else
printf("%d\n", mp[p][m - 1]);
}
return 0;
}