Removed some HashMaps from DAGScheduler by storing information in Stage.

@kayousterhout @markhamstra please take a look ... 
could now do this as `filter { !_.jobIds.contains(jobId) }`