牛客练习赛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都是合法子串。