Query-Focused Retrieval Heads Improve Long-Context Reasoning and Re-ranking