Add the following lines to your $HOME/.bash_profile or $HOME/.bashrc (if you are using zsh then ~/.zprofile or ~/.zshrc) config file:
export JAVA_HOME="/Applications/Android Studio.app/Contents/jre/jdk/Contents/Home"