AC:
if(a[i]<mid) swap(a[i++],a[j++]); else if(a[i]>mid) swap(a[i],a[--k]); else i++;
WA:
if(a[i]<mid) swap(a[i++],a[j++]); if(a[i]>mid) swap(a[i],a[--k]); if(a[i]==mid) i++;