sudo su is deprecated and will indeed mess up your environment.
Use sudo -i instead.
(This also has the advantage that it works on systems without su)
Edit: I forgot something:
Add export PS1="%F{013}%2~%f$(git_super_status)%(?.%F{004}.%F{001}✕%?)$ %f"
at the end of /root/.zshrc