码迷,mamicode.com
首页 > 其他好文 > 详细

思维+贪心+dp——cf1341D

时间:2020-04-28 15:14:29      阅读:56      评论:0      收藏:0      [点我收藏+]

标签:put   col   vector   mes   思维   ack   --   cin   amp   

感觉dp状态还是挺不容易想到的

/*
dp[i][j]表示后i个数用j根火柴棒是否可行 
mp[i][j]表示第i个字符变成j需要多少根火柴棒
求答案时从高位到低位贪心
*/ #include<bits/stdc++.h> using namespace std; #define N 4005 int n,k,dp[N][N]; string s[10]={ "1110111","0010010","1011101","1011011","0111010", "1101011","1101111","1010010","1111111","1111011", }; string t[N]; int mp[N][10];//t[i]->j需要的根数 int main(){ cin>>n>>k; for(int i=1;i<=n;i++){ cin>>t[i]; for(int j=0;j<10;j++){ string a=t[i],b=s[j]; int tot=0; for(int l=0;l<7;l++){ if(a[l]==0 && b[l]==1)tot++; else if(a[l]==1 && b[l]==0)tot-=10000; } mp[i][j]=tot; } } dp[n+1][0]=1; for(int i=n+1;i>1;i--){ for(int j=0;j<=k;j++)if(dp[i][j]){ for(int l=0;l<10;l++)// t[i-1]->j可行 if(mp[i-1][l]>=0) dp[i-1][j+mp[i-1][l]]=1; } } if(dp[1][k]!=1){ puts("-1");return 0; } int now=k; vector<int>ans; for(int i=1;i<=n;i++){//确定第i位 int x; for(int j=9;j>=0;j--)if(mp[i][j]>=0){ if(dp[i+1][now-mp[i][j]]){ now-=mp[i][j]; x=j; break; } } ans.push_back(x); } for(auto x:ans)cout<<x; puts(""); }

 

思维+贪心+dp——cf1341D

标签:put   col   vector   mes   思维   ack   --   cin   amp   

原文地址:https://www.cnblogs.com/zsben991126/p/12793883.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!