A principal indicação terapêutica do lítio é para