A 嘤嘤的签到

 牛客练习赛110_ACM/NOI/CSP/CCPC/ICPC算法编程高难度练习赛_牛客竞赛OJ (nowcoder.com)

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

题目描述

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

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

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

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

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

输入描述:

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

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

输出描述:

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

示例1

输入

3

414

输出

3

说明

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

#include<bits/stdc++.h>
#define LL long long
using namespace std;
int main(){
	int n;
	cin>>n;
	string s;
	cin>>s;
	s = " " + s;
	vector<int> a(100);//初始化长度为100
	LL ans = 0;
	for(int i=1;i<=n;i++){
		a[s[i]] = i;
		ans += i - min(a['1'] , a['4']);
	}
	cout<<ans<<endl;
	return 0;
}

该题用到了枚举。

​ 对每一个位置，从左到右枚举时，维护最后一个1和4的位置，每次枚举计算贡献时，最后一个1和4的位置中较左的那一个往右都是合法的。

​ 例如，041004080，枚举第8个位置时，最后一个1在第3个位置，最后一个4在第6个位置，最左边的位置是3，min(a['1'] , a['4'])即为3，所以i - min(a['1'] , a['4'])等于8-3=5，因此，00408，0408，408，08，8都是合法子串。