2022AcWing寒假算法每日一题之1913. 公平摄影 （很有意思的一道题）

2022AcWing寒假算法每日一题之1913. 公平摄影

题目链接：AcWing1913. 公平摄影

思路：

1.分析题目，发现只有两种情况：

• 只存在一种牛时，求最长的连续区间（采用双指针算法）
• 存在两种牛时，两种牛的数量相同（采用hash表离散化+前缀和操作）

2.当牛为H时，存入pair 的为-1，当牛为G时存入pair 的为1，则当某一区间中两种牛的数量相等时，该区间和为0

3.当有两种牛时，需要hash表离散化，hash表中存入的是坐标，作用是判断序列之间的长度，因为当某一区间和为0时，存入hash表中时会产生hash冲突，即满足2.条件

4.这题很有意思，将前缀和和hash表离散化相结合！

具体代码如下：

#include <iostream>
#include <math.h>
#include <unordered_map>
#include <cstring>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;//存储初始数据
const int N = 1e5+10;
int n;
PII q[N];
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        int x;
        char b;
        cin>>x>>b;
        if(b=='H') q[i]={x,-1};//如果为H，则存入-1
        else q[i]={x,1};//如果为G，则存入1
    }
    sort(q+1,q+1+n);//排序
    unordered_map<int, int> hash;//定义一个哈希表，离散化
    int ans = 0,sum = 0,last = 0;//sum用来求前缀和,last用来双指针操作
    for(int i=1;i<=n;i++)
    {
        if(!hash.count(sum))//如果hash表中无该数据，则存入坐标
            hash[sum]=q[i].first;
        
        sum+=q[i].second;//前缀和操作
        
        if(hash.count(sum))//若hash表中已经有了该数据，则直接计算长度
            ans = max(ans,q[i].first-hash[sum]);
        
        if(i==1||q[i].second!=q[i-1].second)//双指针操作求只有一种牛的最大长度
            last = q[i].first;//更新last
        ans = max(ans,q[i].first-last);
    }
    cout<<ans<<endl;
    return 0;
}