I think it's like withdrawal. Make a list of quick distractions. Conjugate verbs in another language when he pops into your head. Do ridiculous math. I think it just takes time.

I find no contact and lots of contact with other people helps--- you may be stuck since you will still see him in class. You'll get over this.