寒武纪算法开发岗位二面手撕代码题目记录
题目:
给定一个数组,求能组成的和中,最小不可能的和。
例如:[1,2,3,5],最小不可能的和为12。
解答:
就是一个二维动态规划,下面的答案是我面试完下来自己又重新做的,当时没做出来。
主要就是一个二维dp table,然后dp[i][j]表示数组中前i个数,组成和为j是否可能出现。
填表规则:
data为所给数组。
dp[i][j] 有三个来源决定:
- 上一行当前列的值;
- 上一行从当前列往前推data[i],例如,这里第二行data[i]=2;则就是是看dp[i-1][j-data[i]]。
- 本身dp[i][j].
则得到状态转移方程为:
当data[i] > j时:
dp[i][j] = dp[i-1][j] || dp[i-1][j-data[i]] || dp[i][j]
当j >= data[i] 时:(仅仅考虑上一行当前列的数)
dp[i][j] = dp[i-1][j] || dp[i]
初始化:
列数由数组内所有数之和决定,cols = sum,创建数组的时候多留出一个0.所以第二维维sum+1.
行数就是数组的长度。
1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | |
---|---|---|---|---|---|---|---|---|---|---|---|
1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
2 | 0 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
3 | 0 | 0 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
5 | 0 | 0 | 0 | 0 | 1 | 0 | 0 | 0 | 0 | 0 | 0 |
填表:填写第二行
1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | |
---|---|---|---|---|---|---|---|---|---|---|---|
1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
2 | 1 | 1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
3 | 0 | 0 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
5 | 0 | 0 | 0 | 0 | 1 | 0 | 0 | 0 | 0 | 0 | 0 |
填表:填写第二行
这里你可能会疑惑,第三行第6列这里为什么会填1呢?它是由第2行第(6-3)列得来的。
!!注意这里由于是看当前行之前的数,所以不会出现两个相同数相加的情况,比如第2行第3列,这个3是由1+2得到的,而不是直接是3.
1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | |
---|---|---|---|---|---|---|---|---|---|---|---|
2 | 1 | 1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
3 | 1 | 1 | 1 | 1 | 1 | 1 | 0 | 0 | 0 | 0 | 0 |
5 | 0 | 0 | 0 | 0 | 1 | 0 | 0 | 0 | 0 | 0 | 0 |
填表:填写第三行
1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | |
---|---|---|---|---|---|---|---|---|---|---|---|
1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
2 | 1 | 1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
3 | 1 | 1 | 1 | 1 | 1 | 1 | 0 | 0 | 0 | 0 | 0 |
5 | 1 | 1 | 1 | 1 | 1 | 1 | 1 | 1 | 1 | 1 | 1 |
代码:
#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
using namespace std;
/*
该代码是寒武纪的二面面试题:
题目:
给定一个数组,求能组成的和中,最小不可能的和。
例如:[1,2,3,5],最小不可能的和为12。
*/
int minNoSum(vector<int>& data) {
int n = data.size();
if (n == 0) return 0;
if (n == 1) return 1;
int minv = INT_MAX;
int sum = 0;
sort(data.begin(), data.end());
for (int i = 0; i < n; ++i) {
if (data[i] < minv) minv = data[i];
sum += data[i];
}
// dp[i][j]表示前i个数,和为j是否能取到
vector<vector<bool>> dp(n, vector<bool>(sum+1, 0));
for (int i = 0; i < n; ++i) {
int num = data[i];
dp[i][num] = 1;
}
for (int i = 1; i < n; ++i) {
for (int j = 1; j <= sum; ++j) {
if (j >= data[i]) {
dp[i][j] = dp[i-1][j] || dp[i-1][j-data[i]] || dp[i][j];
}
else {
dp[i][j] = dp[i - 1][j] || dp[i][j];
}
}
}
for (int i = 1; i < sum; ++i) {
if (!dp[n - 1][i]) return i;
}
return sum + 1;
}
int main()
{
vector<int> data = {1,2,3,5};
int res = minNoSum(data);
cout << res << endl;
system("pause");
return 0;
}