华为od机试 Python 【静态扫描】

左手的ㄟ右手 2024-03-08 07:52 112阅读 0赞

### 题目 ###

每当我们扫描一个源代码文件，都需要支付一些金币。但是，如果你觉得某个文件后面还会经常出现，那你可以考虑多花一些金币，将它的扫描结果保存起来，这样下次再遇到它时就不用重新支付扫描费用了。

现在，你的挑战是：根据给出的文件序列和每个文件的大小，确定最省钱的策略，使得总花费的金币最少。

输入：  
第一行：保存一个扫描报告的费用（M）  
第二行：连续出现的文件的编号序列  
第三行：每个文件的大小

输出：  
你需要支付的最少金币数量  
示例：

输入：

5  
2 2 2 2 2 5 2 2 2  
3 3 3 3 3 1 3 3 3  
输出：9

### 代码 ###

def min_scan_cost(report_cost, files, sizes):
        file_info = {
        }  # 文件的大小和出现次数
    
        # 统计每个文件的大小和出现次数
        for idx, file in enumerate(files):
            if file not in file_info:
                file_info[file] = {
        'size': sizes[idx], 'count': 1}
            else:
                file_info[file]['count'] += 1
    
        total_cost = 0
    
        # 计算每个文件的费用
        for _, info in file_info.items():
            scan_cost = info['size'] * info['count']  # 重新扫描的费用
            cache_cost = info['size'] + report_cost  # 保存扫描报告的费用
    
            # 选择较小的费用
            total_cost += min(scan_cost, cache_cost)
    
        return total_cost
    
    report_cost = int(input())  
    files = list(map(int, input().split()))  # 连续出现的文件的编号
    sizes = list(map(int, input().split()))  # 每个文件的大小
    
    print(min_scan_cost(report_cost, files, sizes))