#P2323. 第3题-计网实验2.0
          
                        
                                    
                      
        
              - 
          
          
                      1000ms
            
          
                      Tried: 468
            Accepted: 99
            Difficulty: 8
            
          
          
          
                       所属公司 : 
                              华为
                                
            
                        
              时间 :2024年5月15日-暑期实习
                              
                      
          
 
- 
                        算法标签>动态规划          
 
第3题-计网实验2.0
题面描述:
塔子哥在进行计网实验后,又收到了一周的接口板需求。他需要为两台设备配置接口板,使得每台设备的接口板转发能力之和恰好等于设备的整机转发能力。输入包括目标设备的转发能力、客户订购的接口板数量以及接口板的转发能力列表。若能够满足条件,输出两台设备配置的接口板转发能力列表(需按从小到大排列);若无法满足,则输出-1。
思路
由于题目保证如果存在解,那么解一定是唯一的,所以接口板容量之和肯定是整机转发能力的两倍。
我们可以使用动态规划来解决这个问题,设f[i][j]表示前i个接口板能否组成转发能力为j的整机,如果能组成,那么f[i][j]定义为i表示第i个元素为结尾接口板,那么我们有状态转移方程:
f[i][j] = f[i - 1][j] or f[i - 1][j - a[i]]
其中a[i]表示第i个接口板的容量。由于需要记录路径,我们可以使用pre[i][j]来记录f[i][j]的上一个状态。
pre[i][j] = f[i - 1][j - a[i]]
由于每次只需要前一个状态,所以我们可以使用一维的f来进行优化。
最后将符合条件的路径找出,将其分成两部分,分别排序,然后对比输出即可。
题解
为了给两台设备配置接口板,使得每台设备的接口板转发能力之和恰好等于设备的整机转发能力,我们可以利用动态规划来解决这个问题。具体思路如下:
- 
问题分析:
- 输入中给定的目标设备的转发能力为 m,并且客户订购了 n 块接口板。根据题目的描述,如果存在解,那么所有接口板的容量之和必定等于 2m(即两台设备的转发能力总和)。因此,如果这些接口板的总和不等于 2m,直接输出 -1。
 
 - 
动态规划状态定义:
- 我们定义 
f[i][j]表示前 i 个接口板能否组成转发能力为 j 的整机。如果能组成,那么f[i][j]的值为真。我们还需要使用pre[i][j]来记录路径,表示f[i][j]的上一个状态。 
 - 我们定义 
 - 
状态转移方程:
- 状态转移方程为: f[i][j]=f[i−1][j] or f[i−1][j−a[i]]
 - 其中,a[i] 表示第 i 个接口板的容量。
 
 - 
优化:
- 由于我们只需要使用前一个状态,因此可以将 
f数组优化为一维数组,从而减少空间复杂度。 
 - 由于我们只需要使用前一个状态,因此可以将 
 - 
路径回溯:
- 一旦找到能够组成转发能力为 m 的方案,我们需要回溯找到所用的接口板。利用 
pre数组可以帮助我们记录并找出使用了哪些接口板。 
 - 一旦找到能够组成转发能力为 m 的方案,我们需要回溯找到所用的接口板。利用 
 - 
结果输出:
- 将找到的两部分接口板分别排序,然后进行比较,按照题目要求的顺序输出。
 
 
代码
Python
def main():
    import sys
    input = sys.stdin.read
    data = input().split()
    
    m = int(data[0])  # 读取 m
    n = int(data[1])  # 读取 n
    a = list(map(int, data[2:2 + n]))  # 读取数组 a
    # 如果数组元素的总和不等于 2m,则输出 -1 并退出
    if sum(a) != m * 2:
        print(-1)
        return
    # 初始化前驱数组 pre 和动态规划数组 f
    pre = [[-1] * (m + 1) for _ in range(n)]
    f = [-1] * (m + 1)
    f[0] = 0
    # 动态规划过程
    for i in range(n):
        for j in range(m, a[i] - 1, -1):
            if f[j - a[i]] != -1:
                pre[i][j] = f[j - a[i]]
                f[j] = i
    # 如果能找到一个和为 m 的子集
    if f[m] != -1:
        vis = [False] * n
        i, j = m, f[m]
        while i > 0:
            vis[j] = True
            k = a[j]
            j = pre[j][i]
            i -= k
        ans1, ans2 = [], []
        for i in range(n):
            if vis[i]:
                ans1.append(a[i])
            else:
                ans2.append(a[i])
        # 对两个子集进行排序
        ans1.sort()
        ans2.sort()
        # 比较两个子集,确保 ans1 是字典序较小的
        if ans1[0] > ans2[0] or (ans1[0] == ans2[0] and len(ans1) < len(ans2)):
            ans1, ans2 = ans2, ans1
        # 输出两个子集
        print(" ".join(map(str, ans1)))
        print(" ".join(map(str, ans2)))
if __name__ == "__main__":
    main()
Java
import java.util.*;
public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int m = sc.nextInt();
        int n = sc.nextInt();
        int[] a = new int[n];
        for (int i = 0; i < n; ++i) {
            a[i] = sc.nextInt();  // 读取输入的数组元素
        }
        // 如果数组元素的总和不等于 2m,则输出 -1 并退出
        if (Arrays.stream(a).sum() != m * 2) {
            System.out.println(-1);
            return;
        }
        // 初始化前驱数组 pre 和动态规划数组 f
        int[][] pre = new int[n][m + 1];
        for (int[] row : pre) Arrays.fill(row, -1);
        int[] f = new int[m + 1];
        Arrays.fill(f, -1);
        f[0] = 0;
        // 动态规划过程
        for (int i = 0; i < n; ++i) {
            for (int j = m; j >= a[i]; --j) {
                if (f[j - a[i]] != -1) {
                    pre[i][j] = f[j - a[i]];
                    f[j] = i;
                }
            }
        }
        // 如果能找到一个和为 m 的子集
        if (f[m] != -1) {
            boolean[] vis = new boolean[n];
            for (int i = m, j = f[m]; i > 0;) {
                vis[j] = true;
                int k = a[j];
                j = pre[j][i];
                i -= k;
            }
            List<Integer> ans1 = new ArrayList<>();
            List<Integer> ans2 = new ArrayList<>();
            for (int i = 0; i < n; ++i) {
                if (vis[i]) ans1.add(a[i]);
                else ans2.add(a[i]);
            }
            // 对两个子集进行排序
            Collections.sort(ans1);
            Collections.sort(ans2);
            // 比较两个子集,确保 ans1 是字典序较小的
            if (ans1.get(0) > ans2.get(0) || 
                (ans1.get(0).equals(ans2.get(0)) && ans1.size() < ans2.size())) {
                List<Integer> temp = ans1;
                ans1 = ans2;
                ans2 = temp;
            }
            // 输出两个子集
            for (int i = 0; i < ans1.size(); ++i) 
                System.out.print(ans1.get(i) + (i == ans1.size() - 1 ? "\n" : " "));
            for (int i = 0; i < ans2.size(); ++i) 
                System.out.print(ans2.get(i) + (i == ans2.size() - 1 ? "\n" : " "));
        }
    }
}
C++
#include <bits/stdc++.h>
using namespace std;
int main()
{
    int n, m;
    cin >> m >> n;
    vector<int> a(n);
    for (int i = 0;i < n;++ i) cin >> a[i];
    if (accumulate(a.begin(), a.end(), 0) != m * 2) {
    cout << -1 << endl;
    return 0;
    }
    vector<vector<int>> pre(n, vector<int> (m + 1, -1));
    vector<int> f(m + 1, -1);
    f[0] = 0;
    for (int i = 0;i < n;++ i) {
        for (int j = m;j >= a[i];-- j) if (~f[j - a[i]]) {
            pre[i][j] = f[j - a[i]];
            f[j] = i;
        }
    }
    if (~f[m]) {
        vector<bool> vis(n, false);
        for (int i = m, j = f[m];i > 0;) {
            vis[j] = true;
            int k = a[j];
            j = pre[j][i];
            i -= k;
        }
        vector<int> ans1, ans2;
        for (int i = 0;i < n;++ i) {
            if (vis[i]) ans1.push_back(a[i]);
            else ans2.push_back(a[i]);
        }
        sort(ans1.begin(), ans1.end());
        sort(ans2.begin(), ans2.end());
        if (ans1[0] > ans2[0] || (ans1[0] == ans2[0] && ans1.size() < ans2.size())) swap(ans1, ans2);
        for (int i = 0;i < ans1.size();++ i) cout << ans1[i] << " \n"[i == ans1.size() - 1];
        for (int i = 0;i < ans2.size();++ i) cout << ans2[i] << " \n"[i == ans2.size() - 1];
    }
    return 0;
}
javaScript
const rl = require("readline").createInterface({ input: process.stdin });
var iter = rl[Symbol.asyncIterator]();
const readline = async () => (await iter.next()).value;
(async function () {
    let m = parseInt(await readline()); // 读取 m
    let n = parseInt(await readline()); // 读取 n
    let a = (await readline()).split(" ").map(Number); // 读取数组 a
    let sumA = a.reduce((sum, x) => sum + x, 0);
    if (sumA !== 2 * m) {
        console.log(-1);
        rl.close();
        return;
    }
    let pre = Array.from({ length: n }, () => Array(m + 1).fill(-1));
    let f = Array(m + 1).fill(-1);
    f[0] = 0;
    for (let i = 0; i < n; i++) {
        for (let j = m; j >= a[i]; j--) {
            if (f[j - a[i]] !== -1 && f[j] === -1) {
                pre[i][j] = f[j - a[i]];
                f[j] = i;
            }
        }
    }
    if (f[m] === -1) {
        console.log(-1);
        rl.close();
        return;
    }
    let vis = Array(n).fill(false);
    for (let i = m, j = f[m]; i > 0 && j !== -1;) {
        vis[j] = true;
        let k = a[j];
        j = pre[j][i];
        i -= k;
    }
    let ans1 = [], ans2 = [];
    for (let i = 0; i < n; i++) {
        if (vis[i]) ans1.push(a[i]);
        else ans2.push(a[i]);
    }
    ans1.sort((x, y) => x - y);
    ans2.sort((x, y) => x - y);
    if (
        ans1.length > 0 && ans2.length > 0 &&
        (ans1[0] > ans2[0] || (ans1[0] === ans2[0] && ans1.length < ans2.length))
    ) {
        [ans1, ans2] = [ans2, ans1];
    }
    console.log(ans1.join(" "));
    console.log(ans2.join(" "));
    rl.close();
})();
        题目描述
小明做完了上周的计网实验,但是按照一周一个的原则,他又收到了这周的份额,holy damn!
柜式路由器需要配备接口板才可以工作,接口板用于接入用户业务,且接口板转发能力的和不能大于路由器整机的转发能力。当前某客户订购了2台设备和num块接口板。请计算是否存在一种安装方法,使用户选购的接口板,刚好能装到两台设备上,且每台设备配置的口板的转发能力之和,刚好和整机的转发能力相等。
1、设备整机转发能力的单位是Gbps,Gbps是设备单位时间内传输的比特数,代表千兆比特/秒。为了简化问题,规定值为整数,范围为[1,2000]。
2、客户订购的接口板数量num,值的范围[1,200]。
3、接口板容量的单位也是Gbps,比如1010 40 40 100,代表选购了5块接口板,转发能力分别是10Gbps, 10Gbps, 40Gbps, 40Gbps, 100Gbps,接口板转发能力的范围一般为特定枚举值,为了简化问题,规定值为正整数。
输入描述
第一行是目标设备的转发能力。第二行是客户订购的接口板数量num 。第三行是订购的包含num个接口板的转发能力的列表。
输出描述
如果存在满足要求的安装方法,请分两行输出两台设备配置的接口板的转发能力的列表,且要求每台设备的单板,按转发能力从小到大排列。两台设备的单板,第一个单板转发能力小的优先输出。如果第一个单板转发能力相同,那单板数多的优先输出。如果不存在对应的安装方案,则返回-1。用例保证在满足前面条件的情况下,不会有多种不同的结果。
样例一
输入
100
5
40 10 10 40 100
输出
10 10 40 40
100
样例二
输入
100
3
10 10 20
输出
-1