ICPC 2016 - A. Balanced Diet
State the problem in your own words. Focus on the mathematical or algorithmic core rather than repeating the full statement.
Source-first archive entry
This page is built from the copied files in competitive_programming/icpc/2016/A-balanced-diet. Edit
competitive_programming/icpc/2016/A-balanced-diet/solution.tex to update the written solution and
competitive_programming/icpc/2016/A-balanced-diet/solution.cpp to update the implementation.
The website does not replace those files with hand-maintained HTML. It reads the copied source tree during the build and exposes the exact files below.
Problem Statement
Copied statement text kept beside the solution archive for direct reference.
Problem A
Balanced Diet
Time limit: 2 seconds
Every day, Danny buys one sweet from the candy store and eats it. The store has m types of sweets,
numbered from 1 to m. Danny knows that a balanced diet is important and is applying this concept to
his sweet purchasing. To each sweet type i, he has assigned a target fraction, which is a real number fi
(0 < fi ≤ 1). He wants the fraction of sweets of type i among all sweets he has eaten to be roughly
fi . To be more precise, let si denote the number of sweets of type i that Danny has eaten, and
equal toP
let n = m i=1 si . We say the set of sweets is balanced if for every i we have
nfi − 1 < si < nfi + 1.
Danny has been buying and eating sweets for a while and during this entire time the set of sweets
has been balanced. He is now wondering how many more sweets he can buy while still fulfilling this
condition. Given the target fractions fi and the sequence of sweets he has eaten so far, determine how
many more sweets he can buy and eat so that at any time the set of sweets is balanced.
Input
The input consists of three lines. The first line contains two integers m (1 ≤ m ≤ 105 ), which is the
number of types of sweets, and k (0 ≤ k ≤ 105 ), which is the number of sweets Danny has already
eaten.
The second line contains m positive integers a1 , . . . , am . These numbers are proportional to f1 , . . . , fm ,
ai
that is, fi = Pm . It is guaranteed that the sum of all aj is no larger than 105 .
a
j=1 j
The third line contains k integers b1 , . . . , bk (1 ≤ bi ≤ m), where each bi denotes the type of sweet
Danny bought and ate on the ith day. It is guaranteed that every prefix of this sequence (including the
whole sequence) is balanced.
Output
Display the maximum number of additional sweets that Danny can buy and eat while keeping his diet
continuously balanced. If there is no upper limit on the number of sweets, display the word forever.
Sample Input 1 Sample Output 1
6 5 1
2 1 6 3 5 3
1 2 5 3 5
Sample Input 2 Sample Output 2
6 4 forever
2 1 6 3 5 3
1 2 5 3
This page is intentionally left blank.
Editorial
Rendered from the copied solution.tex file. The original TeX source remains
available below.
Key Observations
Write the structural observations that make the problem tractable.
State any useful invariant, monotonicity property, graph interpretation, or combinatorial reformulation.
If the constraints matter, explain exactly which part of the solution they enable.
Algorithm
Describe the data structures and the state maintained by the algorithm.
Explain the processing order and why it is sufficient.
Mention corner cases explicitly if they affect the implementation.
Correctness Proof
We prove that the algorithm returns the correct answer.
Lemma 1.
State the first key claim.
Proof.
Provide a concise proof.
Lemma 2.
State the next claim if needed.
Proof.
Provide a concise proof.
Theorem.
The algorithm outputs the correct answer for every valid input.
Proof.
Combine the lemmas and finish the argument.
Complexity Analysis
State the running time and memory usage in terms of the input size.
Implementation Notes
Mention any non-obvious implementation detail that is easy to get wrong.
Mention numeric limits, indexing conventions, or tie-breaking rules if relevant.
Code
Exact copied C++ implementation from solution.cpp.
#include <bits/stdc++.h>
using namespace std;
namespace {
void solve() {
// Fill in the full solution logic for the problem here.
}
} // namespace
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
solve();
return 0;
}
Source Files
Exact copied source-of-truth files. Edit solution.tex for the write-up and solution.cpp for the implementation.
\documentclass[11pt]{article}
\usepackage[margin=1in]{geometry}
\usepackage[T1]{fontenc}
\usepackage[utf8]{inputenc}
\usepackage{amsmath,amssymb,amsthm}
\usepackage{enumitem}
\title{ICPC World Finals 2016\\A. Balanced Diet}
\author{}
\date{}
\begin{document}
\maketitle
\section*{Problem Summary}
State the problem in your own words. Focus on the mathematical or algorithmic core rather than repeating the full statement.
\section*{Key Observations}
\begin{itemize}[leftmargin=*]
\item Write the structural observations that make the problem tractable.
\item State any useful invariant, monotonicity property, graph interpretation, or combinatorial reformulation.
\item If the constraints matter, explain exactly which part of the solution they enable.
\end{itemize}
\section*{Algorithm}
\begin{enumerate}[leftmargin=*]
\item Describe the data structures and the state maintained by the algorithm.
\item Explain the processing order and why it is sufficient.
\item Mention corner cases explicitly if they affect the implementation.
\end{enumerate}
\section*{Correctness Proof}
We prove that the algorithm returns the correct answer.
\paragraph{Lemma 1.}
State the first key claim.
\paragraph{Proof.}
Provide a concise proof.
\paragraph{Lemma 2.}
State the next claim if needed.
\paragraph{Proof.}
Provide a concise proof.
\paragraph{Theorem.}
The algorithm outputs the correct answer for every valid input.
\paragraph{Proof.}
Combine the lemmas and finish the argument.
\section*{Complexity Analysis}
State the running time and memory usage in terms of the input size.
\section*{Implementation Notes}
\begin{itemize}[leftmargin=*]
\item Mention any non-obvious implementation detail that is easy to get wrong.
\item Mention numeric limits, indexing conventions, or tie-breaking rules if relevant.
\end{itemize}
\end{document}
#include <bits/stdc++.h>
using namespace std;
namespace {
void solve() {
// Fill in the full solution logic for the problem here.
}
} // namespace
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
solve();
return 0;
}