C. Another Problem on Strin

来源:互联网 发布:jquery ajax解析json 编辑:程序博客网 时间:2024/05/10 07:28

time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

A string is binary, if it consists only of characters "0" and "1".

String v is a substring of string w if it has a non-zero length and can be read starting from some position in string w. For example, string "010" has six substrings: "0", "1", "0", "01", "10", "010". Two substrings are considered different if their positions of occurrence are different. So, if some string occurs multiple times, we should consider it the number of times it occurs.

You are given a binary string s. Your task is to find the number of its substrings, containing exactly k characters "1".

Input

The first line contains the single integer k (0 ≤ k ≤ 106). The second line contains a non-empty binary string s. The length of s does not exceed 106 characters.

Output

Print the single number — the number of substrings of the given string, containing exactly k characters "1".

Please do not use the %lld specifier to read or write 64-bit integers in С++. It is preferred to use the cincout streams or the %I64dspecifier.

Sample test(s)
input
1
1010
output
6
input
2
01010
output
4
input
100
01010
output
0
Note

In the first sample the sought substrings are: "1", "1", "10", "01", "10", "010".

In the second sample the sought substrings are: "101", "0101", "1010", "01010".


解题说明:此题要求统计包含k个1的字串的个数,可以用DP的方法来做,个数可以认为与0相关,在k个1之前有多少个0,那么个数就为多少,因为每次取子串可以少取一个0。首先统计1的个数,用num[cnt]表示第cnt个1前面0的个数,那么num[cnt-k]就表示满足条件的k个1之前0的个数,遍历时统计这个值即可。

#include <stdio.h>#include <stdlib.h>#define N 1000000int k, c, d, dp[N + 1] = {1}, i;long long ans;int main (void) {scanf ("%d\n", &k);while (c = getchar (), c != '\n') {if (c == '1') ++d;if (d >= k) ans += dp[d - k];++dp[d];}printf ("%I64d\n", ans);exit (EXIT_SUCCESS);}