Imitation Learning For Combinatorial Optimization And Contact Tracing