题意:
电子钟显示器,一共7个数码管,可以用来表示9个数字。现在有n个这种显示器,并且一些数码管没有未正常工作,现在问正好打开 k k k个数码管,能够显示的最大的数字是多少。
题解:
首先肯定得判断由当前的显示效果变到
0
−
9
0-9
0−9这几个数字分别需要打开多少个数码管,用
d
i
s
[
i
]
[
j
]
dis[i][j]
dis[i][j]来表示第
i
i
i个显示器显示数字
j
j
j需要几步操作,若不能变换到相应数字则为
−
1
-1
−1,那么可以通过
d
p
dp
dp来判断
k
k
k步操作是否能够能把所有的显示器显示正常。
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j]表示后
n
−
i
n-i
n−i个显示器使用
j
j
j次操作是否能显示正常,如果能正常显示,那么
d
p
[
i
]
[
j
]
=
t
r
u
e
dp[i][j]=true
dp[i][j]=true,否则为
f
a
l
s
e
false
false。(这是由于还稍后还需要求出能显示的最大数字)
那么状态转移方程就为:
i n i t : d p [ n ] [ 0 ] = t r u e init: dp[n][0] = true init:dp[n][0]=true
i f d p [ i ] [ j ] = t r u e & & d i s [ i − 1 ] [ d ] ≠ − 1 if \space dp[i][j] = true \ \&\&\ dis[i-1][d] \neq -1 if dp[i][j]=true && dis[i−1][d]=−1
t h e n d p [ i − 1 ] [ j + d i s [ i − 1 ] [ d ] ] = t r u e then \space dp[i-1][j+dis[i-1][d]] = true then dp[i−1][j+dis[i−1][d]]=true
如果
d
p
[
0
]
[
k
]
=
t
r
u
e
dp[0][k] = true
dp[0][k]=true,那么可以正常显示,下一步求出能显示的最大数字。
这一步贪心求解,对于每一个显示器,判断其能显示的最大的数字是多少。
n o w now now是当前还剩的操作数,初始 n o w = k now=k now=k
那么 j j j从 9 9 9遍历到 0 0 0,如果其能变化到数字 j j j并且 d p [ i + 1 ] [ n o w − d i s [ i ] [ j ] ] = t r u e dp[i+1][now-dis[i][j]]=true dp[i+1][now−dis[i][j]]=true,那么当前数字就是能显示的最大结果。
#include <bits/stdc++.h>
using namespace std;
vector<string> digits = {"1110111", "0010010", "1011101", "1011011", "0111010", "1101011", "1101111", "1010010", "1111111", "1111011"};
int main() {
int n, k;
cin >> n >> k;
vector<string> s(n);
vector<vector<int>> dis(n, vector<int>(10));
for (int i = 0; i < n; i++) {
cin >> s[i];
for (int d = 0; d < 10; d++) {
for (int j = 0; j < 7; j++) {
char x = s[i][j];
char y = digits[d][j];
if (x == '1' && y == '0') {
dis[i][d] = -1;
break;
}
if (x == '0' && y == '1') {
++dis[i][d];
}
}
}
}
vector<vector<int>> dp(n+1, vector<int>(k+1));
dp[n][0] = 1;
for (int i = n; i > 0; i--) {
for (int j = 0; j <= k; j++) {
if (dp[i][j]) {
for (int d = 0; d < 10; d++) {
if (dis[i-1][d] != -1 && j+dis[i-1][d] <= k) {
dp[i-1][j+dis[i-1][d]] = 1;
}
}
}
}
}
if (dp[0][k] == 0) {
cout << -1 << '\n';
return 0;
}
for (int i = 0; i < n; i++) {
int now = -1;
for (int d = 9; d >= 0; d--) {
if (dis[i][d] != -1 && k >= dis[i][d] && dp[i+1][k - dis[i][d]]) {
now = d;
k -= dis[i][d];
break;
}
}
cout << now;
}
cout << '\n';
return 0;
}