sol.py
the code