topcoder SRM 619 DIV2 GoodCompanyDivTwo

    int countGood(vector <int> superior, vector <int> workType) {
        int res = 0;
        for(int i = 0 ; i < superior.size(); ++ i){
            set<int> department;
            department.insert(workType[i]);
            int num = 1;
            for(int j = i+1; j < superior.size(); ++ j ){
                if(superior[j] == i){
                    department.insert(workType[j]);
                    num++;
                }
            }
            if(num == department.size()) res++;
        }
        return res;
    }