嘤嘤的签到

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

本来这一场比赛是小白月赛，但是因为嘤嘤的题目越出越难，就只能变成练习赛了呢！但大部分的题目还是小白难度，相信大家都能AK，但是为了以防万一，嘤嘤还是放一题血签吧！

        嘤嘤突然发现，今天好像是一个攻击性很强的日子，除此之外，下个月的今天和上上上个月的今天都是很臭的日子，上上个月的今天又是小木曾雪菜的生日，上个月的今天还是 π\piπ 节。

        世界名画——《你为什么这么熟练啊！》（妮露（左），神里绫华（中），荧（右））（图片加载失败）

        在op414、哼哼啊啊啊、届不到、高等(su)数(tui)学的轮番刺激下，嘤嘤已经对同时出现1、4这两个数字的数字串PTSD了，嘤嘤嘤！

        给定一个数字串，问有多少个连续子串不会让嘤嘤PTSD。

输入描述:

第一行输入一个正整数 n(1≤n≤106)n(1 \le n \le 10^6)n(1≤n≤106) ，表示数字串的长度。

第二行输入一个长度为 nnn 的数字串。

输出描述:

输出一个整数表示不会让嘤嘤PTSD的子串数量。

示例1

输入

复制3 414

3
414

输出

复制3

3

说明

"4","1","4"这三个子串不同时包含1和4。
"41","14","414"这三个子串都同时包含了1和4。
所以答案是3。

#include<stdio.h>
int main()
{
    long long n;
    scanf("%lld",&n);
    getchar();
    char a[1000000];
    gets(a);
    long long sum= 0; //数据规模大 所以sum要用长整型
    int aa=0,bb=0;
    int i=0;
    while(a[i]!='\0')
    {
        if(a[i] == '1'){
            bb++;
            aa=0;     //1 4不能在一起 所以归零
            sum += bb;
            continue;
        }
        else if(a[i] == '4'){
            aa++;
            bb=0;     //1 4不能在一起 所以归零
            sum+=aa; 
            continue;
        }
            aa++;
            bb++;
            if(aa>bb)
            sum+=aa;
            else
            sum+=bb;
    }
printf("%lld",sum);
}