This is a problem from Tao's Analysis I.
We are asked to show that the axiom of choice is equivalent to the statement that for any sets $A$ and $B$ for which a surjection $g:B\to A$ exists, an injection $f:A\to B$ exists.
I have proven that the axiom of choice implies the statement , but I am having difficulty with the other implication. The hint given in the book is to use the previous exercise, which says that the axiom of choice is equivalent to the statement that if $I$ is a set, $\forall\alpha\in I$ $X_\alpha$ is a non-empty-set and $\forall \alpha,\beta\in I, \alpha\neq \beta\implies X_\alpha\cap X_\beta=\emptyset$, then there exists a set $Y$ with $\forall \alpha\in I ,|Y\cap X_\alpha|=1$.
To use the previous exercise we need disjoint sets, so we consider the sets $\{\alpha\}\times X_\alpha$ instead of just $X_\alpha$. I think then we should find a set $Y$ with $\forall \alpha \in I,|Y\cap(\{\alpha\}\times X_\alpha)|=1$, perhaps by using that there is some surjection between two sets and so by assumption some injection which goes the other way, whose image is suitable as a set $Y$. I think that is the idea, but I can't figure out exactly what these functions should be.
If I am incorrect, please correct me. Any help is appreciated.