北邮机试 | bupt oj | 94. 最小距离查询

时间限制 1000 ms 内存限制 65536 KB

题目描述

给定一个由小写字母a到z组成的字符串S，其中第i个字符为S[i]（下标从0开始）。你需要完成下面两个操作：
INSERT c
其中c是一个待输入的字符。你需要在字符串的末尾添加这个字符。保证输入的字符同样是a到z之间的一个小写字母。
QUERY x
其中x是一个输入的整数下标。对于这个询问，你需要回答在S当中和S[x]相等且与x最近的距离。输入保证x在当前字符串中合法。
例如S = "abcaba"，如果我们操作：
INSERT a
则在S的末端加一个字符a，S变成"abcabaa"。
接下来操作
QUERY 0
由于S[0] = a，在S中出现的离他最近的a在下标为3的位置上，距离为3 - 0 = 3。因此应当输出3。
接下来，如果
QUERY 4
S[4] = b，S中离它最近的b出现在下标为1处，距离为4 - 1 = 3。同样应当输出3。
给定初始字符串S和若干操作，对于每个QUERY，你需要求出相应的距离。

HINT 由于输入数据较大，C/C++中推荐使用scanf进行读入以获得更快的读入速度。同时请注意算法复杂度。

输入格式

输入的第一行是一个正整数T(T≤20)，表示测试数据的组数。
每组输入数据的第一行是一个初始串S。第二行是一个正整数m(1≤m≤100000)，表示总共操作的数量。接下来m行，每行表示一个操作。操作的格式如上所述。
数据保证在任何情况下，S的长度不会超过100000。

输出格式

对于每个QUERY，输出所求的最小距离。如果S中其它位置都不存在和它相同的字符，输出-1。

输入样例

2
axb
3
INSERT a
QUERY 0
QUERY 1
explore
3
INSERT r
QUERY 7
QUERY 1

输出样例

3
-1
2
-1

AC代码

//r
#include<bits/stdc++.h>
#define MAXN 100010
#define INF 1e9
using namespace std;
char s[MAXN];
int cnt[27], pre[27];
int dp[MAXN][2];
int c_query(int index){
    int ans = min(dp[index][0],dp[index][1]);
    if(ans !=INF)
        return ans;
    else return -1;
}
void c_insert(char character,int &len){
    s[len] = character;
    int tmp=character -'a';
    cnt[tmp]++;
    if(cnt[tmp]>=2)
    {
        dp[len][0]=dp[pre[tmp]][1]=len-pre[tmp];
    }
    pre[tmp]=len++;
}
void initDis(char s[],int len){
    int tmp;
    for(int i=0;i<len;i++){
        tmp = s[i]-'a';
        cnt[tmp]++;
        if(cnt[tmp]>=2){
            dp[i][0]=dp[pre[tmp]][1]=i-pre[tmp];
        }
        pre[tmp]=i;
    }
    //for(int i=0;i<len;i++){
    //	printf("%d %d\n",dp[i][0],dp[i][1]);
	//} 
}
void init(){
    for(int i=0;i<MAXN;i++)
        dp[i][0]=dp[i][1]=INF;
    for(int i=0;i<27;i++){
        cnt[i]=pre[i]=0;
    }
}
int main()
{
    int t,m,index;
    char command[10],character[10];
    scanf("%d",&t);
    while(t--){
        scanf("%s",s);
        int len = strlen(s);
        init();
        initDis(s,len);
        scanf("%d",&m);
        while(m--){
            scanf("%s",command);
            if(command[0] == 'Q'){
            	scanf("%d",&index);
                int ans = c_query(index);
                printf("%d\n",ans);
            }
            else if(command[0] == 'I'){
            	scanf("%s",character);
                c_insert(character[0],len);
            }
        }
    }
    return 0;
}