2024年5.15日-暑期实习-第三题（300分）-塔子哥的计网实验2.0

在线评测链接

题目描述

\qquad 塔子哥做完了上周的计网实验，但是按照一周一个的原则，他又收到了这周的份额，holy damn！

\qquad 柜式路由器需要配备接口板才可以工作,接口板用于接入用户业务，且接口板转发能力的和不能大于路由器整机的转发能力。当前某客户订购了2台设备和num块接口板。请计算是否存在一种安装方法，使用户选购的接口板，刚好能装到两台设备上，且每台设备配置的口板的转发能力之和,刚好和整机的转发能力相等。

\qquad 1、设备整机转发能力的单位是Gbps,Gbps是设备单位时间内传输的比特数，代表千兆比特/秒。为了简化问题,规定值为整数,范围为[1,2000]。

\qquad 2、客户订购的接口板数量num,值的范围[1,200]。

\qquad 3、接口板容量的单位也是Gbps，比如1010 40 40 100，代表选购了5块接口板,转发能力分别是10Gbps, 10Gbps, 40Gbps, 40Gbps, 100Gbps，接口板转发能力的范围一般为特定枚举值，为了简化问题，规定值为正整数。

输入描述

\qquad 第一行是目标设备的转发能力。第二行是客户订购的接口板数量num 。第三行是订购的包含num个接口板的转发能力的列表。

输出描述

\qquad 如果存在满足要求的安装方法，请分两行输出两台设备配置的接口板的转发能力的列表，且要求每台设备的单板，按转发能力从小到大排列。两台设备的单板，第一个单板转发能力小的优先输出。如果第一个单板转发能力相同，那单板数多的优先输出。如果不存在对应的安装方案，则返回-1。用例保证在满足前面条件的情况下，不会有多种不同的结果。

样例一

输入

100
5
40 10 10 40 100

输出

10 10 40 40
100

样例二

输入

100
3
10 10 20

输出

-1

Limitation

1s, 1024KiB for each test case.

题目思路

由于题目保证如果存在解，那么解一定是唯一的，所以接口板容量之和肯定是整机转发能力的两倍。

我们可以使用动态规划来解决这个问题，设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来进行优化。

最后将符合条件的路径找出，将其分成两部分，分别排序，然后对比输出即可。

代码

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;
}