10221: CCC '24 S5 - Chocolate Bar Partition
[Creator : ]
Description
Maxwell has a chocolate bar that he wants to share with his friends. The chocolate bar can be represented as a $2$ by $N$ array of integers $T_{i,j}$, the tastiness of each square. Maxwell would like to split the entire chocolate bar into connected parts such that the average (mean) tastiness of the chocolate bar is the same for each part. Maxwell would like to know what is the maximum number of connected parts he can split his chocolate bar into as described above.
A part is considered connected if you can visit every cell by moving up, down, left or right.
A part is considered connected if you can visit every cell by moving up, down, left or right.
Input
The first line of input will consist of one positive integer $N$, representing the length of the chocolate bar.
The second line of input contains $N$ spaced integers representing the top row of the chocolate bar where the $j^{th}$ integer from the left represents $T_{1,j}$.
Similarly, the third line of input contains $N$ spaced integers representing the bottom row of the chocolate bar where the $j^{th}$ integer from the left represents $T_{2,j}$.
The second line of input contains $N$ spaced integers representing the top row of the chocolate bar where the $j^{th}$ integer from the left represents $T_{1,j}$.
Similarly, the third line of input contains $N$ spaced integers representing the bottom row of the chocolate bar where the $j^{th}$ integer from the left represents $T_{2,j}$.
Output
Output a single integer, representing the maximum number of connected parts Maxwell can split his chocolate bar into.
Constraints
$1 \leq N \leq 200,000$
$0 \leq T_{i,j} \leq 100,000,000$
$0 \leq T_{i,j} \leq 100,000,000$
Sample 1 Input
2
5 4
6 5
Sample 1 Output
2
An example of how to split this chocolate bar optimally into 2 parts is to have the bottom right corner as its own part and the rest of the chocolate as the second part, as shown below.
Each piece will have an average tastiness of 5.
Each piece will have an average tastiness of 5.
Sample 2 Input
5
1 0 1 2 0
0 2 0 3 1
Sample 2 Output
5
One way to get the optimal split is shown in the following picture:
Note that each piece has an average tastiness of 1.
Note that each piece has an average tastiness of 1.