if(a[i]==1) f[i]=i+1;
if(find(f[i])==i) i++;
else i=f[i];

TLE 40

f[i]=a[i]-1?i:i+1;
i=find(i)-i?f[i]:i+1;

AC