Use topological sort.

let dp[u][c] := the maximum count of vertices with color c of any path starting from vertex u. (by JerryJin2905)