牛客练习赛110——A嘤嘤的签到

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

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

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

        在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(){
    char a[1000000]={'\0'};
    long long int i,m=0,n=0,j=0,sum=0,t,c;
    scanf("%lld",&t);
    scanf("%s",a);
    for(i=0;i<t;i++){
        if(a[i]=='1')m++;
        if(a[i]=='4')n++;
        while(m&&n&&j<i){
            j++;
            if(a[j-1]=='1')m--;
            if(a[j-1]=='4')n--;
        }
        if(!m||!n)sum+=i-j+1;
    }
    printf("%lld",sum);
    return 0;
}