c++ K11734 完美照片

题目描述

科丁博士为了更好宣传科丁星系，想拍一张科丁战士的宣传照片。但是科丁博士想拍一张“完美的”照片，“完美的”是指是照片中的女战士和男战士的数量相等。他让N(1≤N≤50000)个科丁战士占成一条直线，每个战士都有各自的坐标，坐标的范围是0到10^9，同时每个战士用0和1来表示性别，0表示是女战士，1表示是男战士。
请帮助科丁博士计算出一个区间，使这个区间能够达到“完美”，并使得区间尽可能大。区间的大小为区间内最右边的战士的坐标减去最左边的战士的坐标。
输入中，每种性别至少有一名战士，没有两个战士的坐标相同。

输入格式

第一行，一个整数N，表示战士的数量
接下来N行，每行两个整数，分别代表战士i的性别和此战士的坐标

输出格式

1行，一个整数，最大的区间的大小。

输入输出样例

输入样例1                                                            输出样例1：

7                                           11
0 11
1 10
1 25
1 12
1 4
0 13
1 22

说明

【数据规模与约定】：
对于30%的数据，1≤N≤1000
对于50%的数据，1≤N≤10000
对于100%的数据，1≤N≤50000

【样例说明】：

有7个战士，像这样在数轴上。
            1                 1  0  1  0                          1        1
+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
0  1  2  3  4  5  6  7  8  9  10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
其中编号为1(at 11),4(at 12),6(at 13),7(at 22)的这4个战士组成一个“完美”的最大的区间，区间大小为22-11=11个单位长度。

方法：

第一步：数据的读取和存储：将女生的性别标记修改为-1，按照位置从小到大对所有战士进行排队。

第二步：计算前缀和和L[x] ：计算性别前缀和数组sum，并将sum中元素值统一加上N+1。

第三步：计算L[x] ： 枚举前缀和数组，设x =sum[i] ，如果x首次出现，则记录L[x] = i；

第四步：打擂台找最大值， 枚举前缀和数组，设x =sum[i]，如果x不是首次出现，则打擂台：ans=max(ans, p[i]-p[L[x]+1])。

代码思路：

如果区间[l, r]是完美的，那么sum[r] – sum[l-1] == 0，即：sum[r] == sum[l-1]：

 

 那直接出代码了：

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=50005;
struct stu{
    int xb,w;
}s[N];
bool cmp(const stu &x,const stu &y){
    return x.w<y.w;
}
int sum[N],L[N*2],R[N*2];
int main(){
	int n,ans=0;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>s[i].xb>>s[i].w;
        if(s[i].xb==0) s[i].xb=-1;
    }
    sort(s+1,s+1+n,cmp);
    sum[0]=n+1;
    //l-r sum[r]==sum[L-1]
    for(int i=1;i<=n;i++){
        sum[i]=sum[i-1]+s[i].xb;
        int x=sum[i];
        if(x!=n+1&&L[x]==0) L[x]=i;
        else ans=max(ans,s[i].w-s[L[x]+1].w);
    }
    cout<<ans;
    return 0;
}