Description
假发通过了不懈的努力,得到了将军家门锁的密码(一串小写英文字母)。但是假发被十四和猩猩他们盯上了,所以假发需要把密码传递出去。因为假发不想十四他们发现几松门前贴的小纸条就是将军家的密码,所以他加密了密码(新八:听起来有点诡异)。加密方法如下:随机地,在密码中任意位置插入随机长度的小写字符串。
不过,假发相信银桑和他那么多年小学同学,一定能猜中密码是什么的(新八:银桑什么时候成攮夷志士了!!!)。可是,写完了小纸条之后,假发觉得有点长,就想截去头和尾各一段(可以为空),让剩下的中间那一段依然包含真~密码。想着想着,假发就想知道有多少种可行方案。结果在沉迷于稿纸之际,假发被投进了狱门岛(新八:……)。于是,就由你计算了。
Input
两行非空字符串,纯小写英文字母,第一行是加密后的密码,第二行是原密码。
第一行长度不超过300000,第二行不超过200。
Output
一行,有多少种方案。注意:不剪也是一种方案。
Hint
Data Constraint
30%的数据满足第一行长度不超过1000。
Source
YPX
Solution
实际上每次只需要找到最短的满足条件的字符串然后统计两头可用的方案数就可以了
但是直接做会重复计算,那么我们记录上一次第一位的位置a,这一次第一位位置b,最后一位c,每次对答案的贡献就是(b-a)*c
Code
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <iostream>
#include <algorithm>
#include <vector>
#define rep(i, st, ed) for (int i = st; i <= ed; i += 1)
#define drp(i, st, ed) for (int i = st; i >= ed; i -= 1)
#define erg(i, st) for (int i = ls[st]; i; i = e[i].next)
#define fill(x, t) memset(x, t, sizeof(x))
#define max(x, y) ((x)>(y)?(x):(y))
#define min(x, y) ((x)<(y)?(x):(y))
#define abs(x) (x)<(0)?(-(x)):(x)
#define ll long long
#define INF 0x3f3f3f3f
#define N 400001
#define E 1001
char str[N], ptr[N];
int lens, lenp;
int main(void) {
scanf("%s", str);
scanf("%s", ptr);
lens = strlen(str);
lenp = strlen(ptr);
int i = 0, j = 0, a = -1, b = 0;
ll ans = 0;
while (i < lens) {
while (str[i] != ptr[j] && i < lens) {i += 1;}
if (i >= lens) {break;}
if (j == 0) {b = a; a = i;}
i += 1;
j += 1;
if (j >= lenp) {
ans += (a - b) * (lens + 1 - i);
j = 0;
i = a + 1;
}
}
std:: cout << ans << std:: endl;
return 0;
}