Traveling on the Axis 算法训练题目

Traveling on the Axis

原文：https://blog.csdn.net/qq_39826163/article/details/82744035
主要是看过之后加上自己的理解。

题目：
题目大意为：
在[0,n]中每0.5处设置一个红绿灯，0表示红灯，1表示绿灯，如果在4.5处有红灯，要从4到5，就要等1s，给出初始的红绿灯状态，每1s会改变状态，即由红灯变成绿灯，或绿灯变成红灯，并且爬过1，就需要1s的时间
最后要求的是这一段路中在走不同的路经需要的时间之和。
就比如第一个例子：101
从（0，1）所需要的时间为1s，（0，2）所需要的时间为2s，（0，3）所需要的时间为3s，（1，2）所需要的时间为2s，（1，3）所需要的时间为3s，（2，3）所需要的时间为1s，则总的答案为1+2+3+2+3+1=12（s）。
代码：

//根据题意可以发现：1.当上一个灯的状态和下一个灯相同时，则需要2s才能通过，反之只需要1s，而当第一个看见是0时，也是2s。
#include<iostream>
using namespace std;
typedef long long LL;
const int maxn=1e5+5;
char s[maxn];
LL f[maxn];
int main()
{
    int T;
    scanf("%d",&T); //输入T个案例
    while(T--) //解决答案
    {
        memset(s,0,sizeof(s)); //将s数组清零
        memset(f,0,sizeof(f)); //将f数组清零
        scanf("%s",s); //输入字符串，因为本题不知道一行有多少个所以用字符串输入
        LL n=strlen(s),ans=0; //计算字符串内有多少字符
        f[1]=1;
        for(LL i=1;i<n;i++) //判断前后两个数组是否为相同的数字
        {
            if(s[i]==s[i-1])f[i+1]=2;
            else f[i+1]=1;
        }
        for(LL i=1;i<=n;i++)
        {
            ans+=i*(n-i+1)*f[i]; //i*(n-i+1)是表示这一阶段总共会出现的次数，n-i+1表示的是这个阶段被当作是初始阶段也就是开始阶段的次数，i*(n-i+1)*f[i]表示f[i]在所有答案中出现的次数
            if(s[i-1]=='0')ans+=n-i+1; //若s[i-1]=0时，说明初始时需要再等1s
            if(f[i]==2)ans-=n-i+1; //如果发f[i]==2时，我们把这个阶段的被当作开始阶段的次数给算进去了，所以需要剪掉多出来的1s；
        }
        printf("%lld\n",ans);
    }
    return 0;
}