From a pragmatic standpoint, Kasich would be a possible because this would probably ensure that Trump would win Ohio. You risk though losing a bit of conservative enthusiasm with Kasich which you could re-ignite by floating Cruz’s name for the Supreme Court.
Don’t think Trump would pick someone just because they would help him win a state.
Would probably be good to ascertain what Trump expects from his VP before attempting to guess who he may pick.