寒假每日一题-1 孤独的照片

Farmer John 最近购入了 NN 头新的奶牛，每头奶牛的品种是更赛牛（Guernsey）或荷斯坦牛（Holstein）之一。

奶牛目前排成一排，Farmer John 想要为每个连续不少于三头奶牛的序列拍摄一张照片。

然而，他不想拍摄这样的照片，其中只有一头牛的品种是更赛牛，或者只有一头牛的品种是荷斯坦牛——他认为这头奇特的牛会感到孤立和不自然。

在为每个连续不少于三头奶牛的序列拍摄了一张照片后，他把所有「孤独的」照片，即其中只有一头更赛牛或荷斯坦奶牛的照片，都扔掉了。

给定奶牛的排列方式，请帮助 Farmer John 求出他会扔掉多少张孤独的照片。

如果两张照片以不同位置的奶牛开始或结束，则认为它们是不同的。

输入格式

输入的第一行包含 NN。

输入的第二行包含一个长为 NN 的字符串。如果队伍中的第 ii 头奶牛是更赛牛，则字符串的第 ii 个字符为 G。否则，第 ii 头奶牛是荷斯坦牛，该字符为 H。

输出格式

输出 Farmer John 会扔掉的孤独的照片数量。

数据范围

3≤N≤5×1053≤N≤5×105

输入样例：

5
GHGHG

输出样例：

3

样例解释

这个例子中的每一个长为 33 的子串均恰好包含一头更赛牛或荷斯坦牛——所以这些子串表示孤独的照片，并会被 Farmer John 扔掉。

所有更长的子串（GHGH、HGHG 和 GHGHG）都可以被接受。

错误代码

时间复杂度为o(n*n),如果数据过大导致TLE。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int main()
{
    int n;
    string s;
    int sum=0;
    cin>>n>>s;
   for(int i=0;i<n;i++)
    {   string t;
        for(int j=i;j<n;j++)
        {
            
                t+=s[j];
                
            if(t.size()>=3)
            { 
            int g=0;
              int h=0;
                for(int i=0;i<t.size();i++)
                {
                    if(t[i]=='G')g++;
                    if(t[i]=='H')h++;
                    
                }if(g==1||h==1)sum++;
                
            }
        }
        
    }
    cout<<sum;
    return 0;
}

正确代码

时间复杂度为o(n)。

整体思路：

遍历整个字符数组，用l[N]来记录每个字符的左侧不同于自身的连续字符数，用r[N]来记录每个字符的右侧不同于自身的连续字符数。

经观察，所有可能的情况数为

1.l[i]*r[i];

2.l[i]-1;

3.r[i]-1;

#include<iostream>
#include<cstdio>
#include <algorithm>

using namespace std;
typedef long long LL;
int n;
const int N = 500010;
char str[N];
int l[N],r[N];

int main()
{
    
    cin>>n>>str;
    for(int i=0,h=0,g=0;i<n;i++)
        if(str[i]=='G')l[i]=h,h=0,g++;
        else l[i]=g,g=0,h++;
    
    for(int i=n-1,h=0,g=0;i>=0;i--)
        if(str[i]=='G')r[i]=h,h=0,g++;
        else r[i]=g,g=0,h++;
        
        
    LL res=0;    
        
    for(int i=0;i<n;i++)
         res+=(LL)l[i]*r[i]+max(l[i]-1,0)+max(r[i]-1,0);
    
   cout<<res;
    return 0;
}