Easy之1303.求团队人数

缺乏、安全感 2022-10-11 01:33 127阅读 0赞

### 员工表：Employee ###

<table> 
 <thead> 
  <tr> 
   <th>Column Name</th> 
   <th>Type</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>employee_id</td> 
   <td>int</td> 
  </tr> 
  <tr> 
   <td>team_id</td> 
   <td>int</td> 
  </tr> 
 </tbody> 
</table>

employee\_id 字段是这张表的主键，表中的每一行都包含每个员工的 ID 和他们所属的团队。

### 问题 ###

编写一个 SQL 查询，以求得每个员工所在团队的总人数。

查询结果中的顺序无特定要求。

### 示例 ###

查询结果格式示例如下：

Employee Table:

<table> 
 <thead> 
  <tr> 
   <th>employee_id</th> 
   <th>team_id</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>1</td> 
   <td>8</td> 
  </tr> 
  <tr> 
   <td>2</td> 
   <td>8</td> 
  </tr> 
  <tr> 
   <td>3</td> 
   <td>8</td> 
  </tr> 
  <tr> 
   <td>4</td> 
   <td>7</td> 
  </tr> 
  <tr> 
   <td>5</td> 
   <td>9</td> 
  </tr> 
  <tr> 
   <td>6</td> 
   <td>9</td> 
  </tr> 
 </tbody> 
</table>

Result table:

<table> 
 <thead> 
  <tr> 
   <th>employee_id</th> 
   <th>team_size</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>1</td> 
   <td>3</td> 
  </tr> 
  <tr> 
   <td>2</td> 
   <td>3</td> 
  </tr> 
  <tr> 
   <td>3</td> 
   <td>3</td> 
  </tr> 
  <tr> 
   <td>4</td> 
   <td>1</td> 
  </tr> 
  <tr> 
   <td>5</td> 
   <td>2</td> 
  </tr> 
  <tr> 
   <td>6</td> 
   <td>2</td> 
  </tr> 
 </tbody> 
</table>

ID 为 1、2、3 的员工是 team\_id 为 8 的团队的成员，  
ID 为 4 的员工是 team\_id 为 7 的团队的成员，  
ID 为 5、6 的员工是 team\_id 为 9 的团队的成员。

来源：力扣（LeetCode）  
链接：https://leetcode-cn.com/problems/find-the-team-size  
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

### 解答 ###

##### 法一：相关子查询 #####

select employee_id,
        (select count(*) 
        from employee e2 
        where e1.team_id=e2.team_id) as team_size
    from employee e1
    order by employee_id

##### 法二：自联结 #####

select e1.employee_id, count(*) as team_size
    from employee e1 join employee e2 using(team_id)
    group by e1.employee_id
    order by e1.employee_id

这里的using(team\_id)等价于on e1.team\_id = e2.team\_id

##### 法三：窗口函数 #####

SELECT
        employee_id,
        COUNT(employee_id) OVER(PARTITION BY team_id) AS team_size
    FROM Employee
    ORDER BY employee_id;

窗口函数：

1.  基本语法：  
    <窗口函数> over (partition by <用于分组的列名>  
    order by <用于排序的列名>)
2.  <窗口函数>的位置，可以放以下两种函数：  
    （1） 专用窗口函数，包括后面要讲到的rank, dense\_rank, row\_number等专用窗口函数。  
    （2） 聚合函数，如sum. avg, count, max, min等
3.  因为窗口函数是对where或者group by子句处理后的结果进行操作，所以窗口函数原则上只能写在select子句中。