I dont know what vai used...i remember reading that he got violin music and sax music and used it a lot for sight reading and whatnot...
but...I know of two books that are great for any guitarist...beginning to advanced....the two by
john petrucci...wild stringdom and rock discipline. *Those are great!