import re
from functools import lru_cache

TEMPLATE_BASE = """//PREPEND BEGIN
{}
//PREPEND END

//TEMPLATE BEGIN
{}
//TEMPLATE END

//APPEND BEGIN
{}
//APPEND END"""

@lru_cache(maxsize=100)
def parse_problem_template(template_str):
    prepend = re.findall(r"//PREPEND BEGIN\\n([\\s\\S]+?)//PREPEND END", template_str)
    template = re.findall(r"//TEMPLATE BEGIN\\n([\\s\\S]+?)//TEMPLATE END", template_str)
    append = re.findall(r"//APPEND BEGIN\\n([\\s\\S]+?)//APPEND END", template_str)
    return {"prepend": prepend[0] if prepend else "",
            "template": template[0] if template else "",
            "append": append[0] if append else ""}

@lru_cache(maxsize=100)
def build_problem_template(prepend, template, append):
    return TEMPLATE_BASE.format(prepend, template, append)