Codeforces Raif Round 1 (Div. 1 + Div. 2)
F
Fruit Sequences
Codeforces
2400
原题
← EG1 →
翻译中…

Zookeeper is buying a carton of fruit to feed his pet wabbit. The fruits are a sequence of apples and oranges, which is represented by a binary string $s_1s_2\ldots s_n$ of length $n$. $1$ represents an apple and $0$ represents an orange.

翻译中…

Since wabbit is allergic to eating oranges, Zookeeper would like to find the longest contiguous sequence of apples. Let $f(l,r)$ be the longest contiguous sequence of apples in the substring $s_{l}s_{l+1}\ldots s_{r}$.

翻译中…

Help Zookeeper find $\sum_{l=1}^{n} \sum_{r=l}^{n} f(l,r)$, or the sum of $f$ across all substrings.

翻译中…

Input

翻译中…

The first line contains a single integer $n$ $(1 \leq n \leq 5 \cdot 10^5)$.

翻译中…

The next line contains a binary string $s$ of length $n$ $(s_i \in \{0,1\})$

翻译中…

Output

翻译中…

Print a single integer: $\sum_{l=1}^{n} \sum_{r=l}^{n} f(l,r)$.

翻译中…

Note

翻译中…

In the first test, there are ten substrings. The list of them (we let $[l,r]$ be the substring $s_l s_{l+1} \ldots s_r$):

翻译中…
  • $[1,1]$: 0
  • $[1,2]$: 01
  • $[1,3]$: 011
  • $[1,4]$: 0110
  • $[2,2]$: 1
  • $[2,3]$: 11
  • $[2,4]$: 110
  • $[3,3]$: 1
  • $[3,4]$: 10
  • $[4,4]$: 0
翻译中…

The lengths of the longest contiguous sequence of ones in each of these ten substrings are $0,1,2,2,1,2,2,1,1,0$ respectively. Hence, the answer is $0+1+2+2+1+2+2+1+1+0 = 12$.

翻译中…

样例

Input #1
4
0110
Output #1
12
InputCopy OutputCopy
Input #2
7
1101001
Output #2
30
InputCopy OutputCopy
Input #3
12
011100011100
Output #3
156
InputCopy OutputCopy
题目信息
题号
1428F
时间限制
2 s
内存限制
256 MB
标签
二分
数据结构
分治
动态规划
双指针

题面与样例版权归 Codeforces 及出题人所有;中文翻译由 AI 生成(CI 管线),如有出入请以原文为准。