pi-Flow: Policy-Based Few-Step Generation via Imitation Distillation