Removed some HashMaps from DAGScheduler by storing information in Stage.

@kayousterhout @markhamstra please take a look ... 
nit: It's a Set not a List