HDU 6345：子串查询（前缀和）

子串查询

Time Limit: 3500/3000 MS (Java/Others)    Memory Limit: 262144/262144 K (Java/Others)
Total Submission(s): 633    Accepted Submission(s): 301

Problem Description

度度熊的字符串课堂开始了！要以像度度熊一样的天才为目标，努力奋斗哦！
为了检验你是否具备不听课的资质，度度熊准备了一个只包含大写英文字母的字符串 A[1,n]=a1a2⋯an，接下来他会向你提出 q 个问题 (l,r)，你需要回答字符串 A[l,r]=alal+1⋯ar 内有多少个非空子串是 A[l,r] 的所有非空子串中字典序最小的。这里的非空子串是字符串中由至少一个位置连续的字符组成的子序列，两个子串是不同的当且仅当这两个子串内容不完全相同或者出现在不同的位置。

记 |S| 为字符串 S 的长度，对于两个字符串 S 和 T ，定义 S 的字典序比 T 小，当且仅当存在非负整数 k(≤min(|S|,|T|)) 使得 S 的前 k 个字符与 T 的前 k个字符对应相同，并且要么满足 |S|=k 且 |T|>k，要么满足 k<min(|S|,|T|) 且 S 的第 k+1 个字符比 T 的第 k+1 个字符小。例如 "AA" 的字典序比 "AAA" 小，"AB" 的字典序比 "BA" 小。

Input

第一行包含一个整数 T，表示有 T 组测试数据。
接下来依次描述 T 组测试数据。对于每组测试数据：
第一行包含两个整数 n 和 q，表示字符串的长度以及询问的次数。
第二行包含一个长为 n 的只包含大写英文字母的字符串 A[1,n]。
接下来 q 行，每行包含两个整数 li,ri，表示第 i 次询问的参数。
保证 1≤T≤10，1≤n,q≤105，1≤li≤ri≤n。

Output

对于每组测试数据，先输出一行信息 "Case #x:"（不含引号），其中 x 表示这是第 x 组测试数据，接下来 q 行，每行包含一个整数，表示字符串 A[l,r] 中字典序最小的子串个数，行末不要有多余空格。

Sample Input

1
2 3
AB
1 1
1 2
2 2

Sample Output

Case #1:
1
1
1

思路

就是求区间内最小的字母出现的次数，用二维数组记录每个字母在当前位置出现的次数，然后对于每次询问，直接遍历26个字母查找就行了。第一次因为循环找最小的字母写错循环了，TLE

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <limits.h>
#include <map>
#include <stack>
#include <queue>
#include <vector>
#include <set>
#include <string>
#define ll long long
#define ms(a) memset(a,0,sizeof(a))
#define pi acos(-1.0)
#define INF 0x3f3f3f3f
const double E=exp(1);
const int maxn=1e5+10;
using namespace std;
char ch[maxn];
int a[30][maxn];
int main(int argc, char const *argv[])
{
	ios::sync_with_stdio(false);
	int t;
	int n,q;
	int x=0;
	scanf("%d",&t);
	while(t--)
	{
		ms(a);
		scanf("%d%d",&n,&q);
		scanf("%s",ch);
		for(int i=0;i<n;i++)
		{
			// 处理前缀和
			for(int j=0;j<26;j++)
				a[j][i]=a[j][i-1];
			a[ch[i]-'A'][i]+=1;
		}
		printf("Case #%d:
",++x);
		int l,r;
		while(q--)
		{
			int ans=0;
			scanf("%d%d",&l,&r);
			for(int i=0;i<26;i++)
			{
				// 如果两端点的该字母前缀和不同
				if(a[i][r-1]-a[i][l-2])
				{
					ans=a[i][r-1]-a[i][l-2];
					break;
				}
			}
			printf("%d
",ans);
		}
	}
	return 0;
}