This seems like a relatively simple indexing problem to me. Why not just use another data structure to store the association for fast querying?