[Usaco2014 Open]Fair Photography

[Usaco2014 Open]Fair Photography

Time Limit: 1 Sec Memory Limit: 128 MB
Description
农夫John（以后简称“FJ”）的N头奶牛站在一个一维长篱笆上的不同位置 (1 <= N <= 100,000)。第i头奶牛站在
位置x_i (一个处于范围0…1,000,000,000的整数)，其品种为b_i（要么为‘G’，表示根西种Guernsey，要么为
‘H’，表示霍斯坦种Holstein）。每头奶牛的位置各不相同。FJ想要为一段连续区间内的奶牛拍张照片，用来参
加郡里的展览会，但是他想让照片里各个品种奶牛的数量是“公平的”。所以，他想要使得无论照片里出现了多少
个品种的奶牛，这些品种奶牛的数量必须相同（例如，一张照片里可以全是霍斯坦种奶牛，也可以出现27头根西种
奶牛和27头霍斯坦种奶牛，但不能出现10头霍斯坦种奶牛和9头根西种奶牛，这种情况下各个品种数量不同）。帮
助FJ拍一张满足其要求的最大尺寸照片。一张照片的尺寸为照片中最左边的奶牛和最右边的奶牛的位置差。可能FJ
拍出的照片里只有一头奶牛，此时照片的尺寸为0。
Input
第1行：整数N。
第2…1+N行：第i+1行包含x_i和b_i。
Output
第1行：一个整数，表示一张“公平的”照片的最大尺寸。
Sample Input
6
4 G
10 H
7 G
16 G
1 G
3 H
【输入说明】
有6头奶牛（从左到右）品种分别为G, H, G, G, H, G。
Sample Output
7
【输出说明】
FJ可以拍出的最大尺寸“公平的”照片是出现中间4头奶牛的照片，包括了2头霍斯坦种奶牛和2头根西种奶牛。

解析

这个题暴搜是绝对会超时的，我们就可以用前缀和，全是一样的就特判一下就可以了。

代码如下

#include<bits/stdc++.h>
using namespace std;
struct we{
	int w,z;
}a[200001];
int f[200001],tot[200001];
int cmp(we x,we y)
{
	return x.w<y.w;
}
int main()
{
	int n,sum=0;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
	 char b;
     scanf("%d",&a[i].w);
     cin>>b;
     if(b=='G')a[i].z=-1;
     else a[i].z=1;
     if(b=='G')sum++;
    }
    sort(a+1,a+n+1,cmp);
    if(sum==n||sum==0||sum==n/2)
    {
    	cout<<a[n].w-a[1].w;
    	return 0;
	}
	int ans=0,left=1,right=1;
	for(int i=2;i<=n;i++)
	{
		if(a[i].z!=a[i-1].z)
	    {
	     int right=i-1;
		 if(ans<a[right].w-a[left].w)ans=a[right].w-a[left].w;
		 left=i;
		 }
	}//为全为一种的特判一下 
    tot[0]=100000;
    for(int i=1;i<=n;i++)
    tot[i]=tot[i-1]+a[i].z;
    for(int i=1;i<=n;i++)
    {
	if(!f[tot[i]])
       f[tot[i]]=a[i+1].w;	
	else 
	   ans=max(ans,a[i].w-f[tot[i]]);
    } //用前缀和 
    cout<<ans;
    return 0;
}