В общем случае уравнение Шрёдингера записывается в следующем виде:
H*w=E*w,
здесь Н - оператор энергии, Е - собственно энергия.
Записывайте теперь полную энергию системы, подставляйте вместо импульса его оператор и загоняйте это всё в уравнение.