Removed some HashMaps from DAGScheduler by storing information in Stage.

@kayousterhout @markhamstra please take a look ... 
btw i'm confused by this actually. This function accepts an ActiveJob, which already has a link to the final stage. Why do we also accept resultStage? @kayousterhout @markhamstra