1145. 关联查询

1145. 关联查询

中文English

给定一张员工表datalist1，存储员工ID，员工姓名。
给定一张员工工作时长表datalist2，存储员工ID，月份，工时。
计算每个员工1-3月每月工时及总工时。

输入描述：
[[员工ID,员工姓名],[员工ID,员工姓名],...]
[[员工ID,月份,工时,月份,工时,月份,工时],[员工ID,月份,工时,月份,工时,月份,工时],...]

输出描述：
[[员工姓名,一月份工时，二月份工时，三月份工时，总工时],[员工姓名,一月份工时，二月份工时，三月份工时，总工时],...]

样例

输入: 
[["1","zhangwei01"]]
[["1","01","200","02","150","03","196"]]
输出: 
[["zhangwei01","200","150","196","546"]]

注意事项

两张表均按员工ID从小到大依次给出相关数据，返回的表中的数据也应按员工ID从小到大依次排列。

 

 

输入测试数据 (每行一个参数)如何理解测试数据？

class Solution:
    """
    @param datalist1: a list represents the employee table
    @param datalist2: a list represents the employee hours table
    @return: Returns a list of strings represents the datalist3
    """
    def getList(self, datalist1, datalist2):
        # write your code here
        #初始化
        res = []
        
        l = len(datalist1)
        for i in range(l):
            total_times = int(datalist2[i][2]) + int(datalist2[i][4]) + int(datalist2[i][6])
            c = [datalist1[i][1],datalist2[i][2],datalist2[i][4],datalist2[i][6],str(total_times)]
            
            res.append(c)
        
        return res