题目
参考
给定
0
−
9
0-9
0−9这些数字的电子表示(7根帮子组成),给出n个电子数,添加k个棒子上去,使得最终的数最大。不能构造出数,输出-1。最终的数允许前缀0,但保持总个数为
n
n
n。
思路:预处理,从低位往前标记,将可达状态置1;完毕后
c
h
e
c
k
check
check最终是否有数存在,没有输出-1,否则从高位往低位枚举,每次贪心选择最大数。
#include<bits/stdc++.h>
using namespace std;
const int maxn = 2010;
const string Num[10] = {"1110111", "0010010", "1011101", "1011011", "0111010", "1101011", "1101111", "1010010", "1111111", "1111011"};
int num[10];
int a[maxn];
int n,k;
char s[10];
void init() {
for(int i = 0;i < 10;++i)
for(int j = 0;j < 7;++j)
if(Num[i][j] == '1')
num[i] |= 1<<j;
}
bool vis[maxn][maxn];
char ans[maxn];
int main() {
init();
scanf("%d%d",&n,&k);
for(int i = 0;i < n;++i) {
scanf("%s",s);
for(int j = 0;j < 7;++j)
if(s[j] == '1')
a[i] |= 1<<j;
}
vis[n][0] = 1;
for(int i = n-1;i >= 0;--i)
for(int j = 9;j >= 0;--j)
if((num[j]&a[i]) == a[i]) {
int diff = __builtin_popcount(num[j]^a[i]);
for(int x = diff;x <= k;++x)
vis[i][x] |= vis[i+1][x-diff];
}
if(!vis[0][k]) {
printf("-1\n");
return 0;
}
for(int i = 0;i < n;++i)
for(int j = 9;j >= 0;--j)
if((num[j]&a[i]) == a[i]) {
int diff = __builtin_popcount(num[j]^a[i]);
if(diff <= k && vis[i+1][k-diff]) {
ans[i] = j+'0';
k -= diff;
break;
}
}
ans[n] = '\0';
printf("%s\n",ans);
}