Sicily.1063. Who's the Boss(排序,确定上下属关系)

来源:互联网 发布:linux 启动jar包脚本 编辑:程序博客网 时间:2024/06/08 09:03
/*1063. Who's the Boss  题目大意:给出员工ID,收入,身高。 当a收入大于b且身高大于b,            且a的收入高于b的量是最小时,a为b的直接上司            b的下属也是a的下属。            对查询某个ID,输出其直属上司以及该ID的下属数量  方法:1.先把员工按照收入从大到小降序排列。        2.从底部开始往上遍历数组,依次找出每个员工的直属上司,将自身          fatherID = 上司所处数组下标i           并且初始化下属数量childNum = 0, 同时在相应上司的child          集合中push_back(下属所处数组下标i)          当没有上司时,其fatherID = -1;           3、再从底部往上遍历数组,当发现其下属容器chilid不为空时,           遍历容器,把其下属的childNum+1 加到自身的childNum上        4、针对每一个查询的ID,找出相应的ID后输出结果         */#include <stdlib.h>#include <iostream>#include <algorithm>#include <stdio.h>#include <vector>using namespace std;struct Employee{  int ID, salary, height, fatherID, childNum;   vector<int> child;  };bool compare(Employee a, Employee b){      return a.salary > b.salary ;    }int main (){    int testN;    int employeeNum, queryNum;    scanf("%d", &testN);    //cin >> testN;    Employee employee[30000];    int query[200];    int ID, salary, height;    for(int i=1; i<=testN; i++){       //cin >> employeeNum >> queryNum;       scanf("%d %d", &employeeNum,&queryNum);             for(int k=0; k<employeeNum; k++){          //cin >> employee[k].ID >> employee[k].salary >> employee[k].height;          scanf("%d %d %d", &employee[k].ID, &employee[k].salary , &employee[k].height);          employee[k].childNum = 0;          employee[k].child.clear();          }       sort(employee,employee+employeeNum, compare);                     for(int j=employeeNum-1; j>=0; j--){          int k;          for(k=j-1; k>=0; k--){             if(employee[k].height>= employee[j].height)                break;                        }          if(k>=0){             employee[j].fatherID = k;             employee[k].child.push_back(j);            }          else             employee[j].fatherID  = -1;            }       for(int j=employeeNum-1; j>=0; j--){          if(employee[j].child.size() >0){            for(int k=0; k <employee[j].child.size(); k++ ){                 employee[j].childNum += employee[employee[j].child[k]].childNum + 1;              }                               }         }          for(int k=0; k<queryNum; k++)       {          int queryID ;          //cin >> queryID;          scanf("%d", &queryID);          int j;          for(j=0; j<employeeNum; j++)            if(employee[j].ID == queryID)                break;          if(employee[j].fatherID == -1)               //cout << "0 " << employee[j].childNum << endl;               printf("0 %d\n", employee[j].childNum);              else             //cout << employee[employee[j].fatherID].ID << " " << employee[j].childNum << endl;             printf("%d %d\n", employee[employee[j].fatherID].ID,employee[j].childNum );             }           }        system("pause");    return 0;    }

原创粉丝点击