Removed some HashMaps from DAGScheduler by storing information in Stage.

@kayousterhout @markhamstra please take a look ... 
Why not just do this as part of the Stage constructor?