3175. 人物相关性分析——第十届蓝桥杯省赛C++C组,第十届蓝桥杯省赛JAVAB/C组

题目链接

题解

• 题目问我们，有多少个Alice和Bob同时出现。对于每一个Alice和他同时出现的Bob的个数，设此Alice的位置是p,准确的说’A’的位置是p，就是在[p-k-3,p+k+5]范围内Bob的个数
• 很容易想到暴力枚举每一个Alice，然后在她的左右去寻找有多少个Bob,但是这样会超时
• 我们首先记录下每一个Alice和每一个Bob的位置，分别用a和b数组存储
• 然后我们用双指针lp和rp维护，大小为[p-k-3,p+k+5)左闭右开的窗口，那么对于每一个Alice，和他同时出现的Bob个数就是rp-lp
• 这道题在寻找每一个Alice和Bob出现的位置时，要注意前后是不是字母等等细节，我是用一个函数完成的

AC代码

#include<bits/stdc++.h>
using namespace std;
#define mem(a,b) memset(a,b,sizeof a)
#define PII pair<int,int>
#define ll long long
#define ull unsigned long long
#define IOS ios::sync_with_stdio(0),cin.tie(0)
#define gcd(a,b) __gcd(a,b)
#define ft first
#define sd second
#define endl '\n'
#define PI acos(-1.0)
#define lcm(a,b) a/gcd(a,b)*b
#define INF_INT 0x3f3f3f3f
#define INF_LONG 4557430888798830399
const int N = 4e6+9;
int ta,tb;
int a[N],b[N];
string s;
int  solve(string ss,int *cnt)//求字符串ss在串s中出现的位置
{
	int t=0,n=s.size();
	for(int i=0; i<n; i++)
	{
		if(s[i]==ss[0])
		{
			if(i)//要注意判断前面是不是字母
			{
				char ch=s[i-1];
				if((ch>='a'&&ch<='z')||(ch>='A'&&ch<='Z')) continue; 
			}
			int flag=1,j;
			string tmp="";
			for(j=i; j<n&&j-i<ss.size(); j++)
			{
				tmp+=s[j];
			}
			if(tmp!=ss) flag=0;
			//要注意判断后面是不是字母
			if(j<n&&((s[j]>='a'&&s[j]<='z')||(s[j]>='A'&&s[j]<='Z'))) flag=0;
			if(flag)  cnt[t++]=i;
		}
	}
	return t;
} 
int main()
{
	int k;
	cin>>k;
	getchar();
	getline(cin,s);
	int ta=solve("Alice",a);
	int tb=solve("Bob",b);
	ll ans=0;
	for(int i=0,lp=0,rp=0; i<ta; i++)
	{
		while(b[lp]<a[i]-3-k) lp++;
		while(b[rp]<=a[i]+k+5&&rp<tb) rp++;
		//cout<<lp<<" "<<rp<<endl;
	    if(rp-lp>0) ans+=(ll)(rp-lp);
	} 
	cout<<ans<<endl;
	return 0;
 }