题意:找到第一个比n大的幸运数,幸运数的定义是只由4或7组成,且4的个数等于7。
开始我的做法是暴力,按照每位贪心的找,坑点和要考虑的地方很多,写的很麻烦。
#include
using namespace std;
const int MAX=100;
char s[MAX];
char ss[MAX];
int T;
bool check(char *s){
int len=strlen(s);
int num1=0,num2=0;
for(int i=0;iif(s[i]=='4') num1++;
if(s[i]=='7') num2++;
}
if(num1==num2&&num1+num2==len) return true;
return false;
}
int main(){
cin>>T;
while(T--){
scanf("%s",s);
if(check(s)){
printf("%s\n",s);
continue;
}
int len=strlen(s);
if(len%2==1){
int n=(len+1)/2;
for(int i=0;iprintf("4");
for(int i=0;iprintf("7");
printf("\n");
}
else{
int n=len/2;
for(int i=0;i'7';
for(int i=0;i'4';
ss[len]=0;
if(strcmp(ss,s)<0){
n++;
for(int i=0;iprintf("4");
for(int i=0;iprintf("7");
printf("\n");
}
else{
int pos=0;
int flag=0;
int f=0;
int n1=0,n2=0;
for(int i=0;iif(s[i]<'4'){
if(n1'4';
else if(n1==n) ss[i]='7';
flag=i+1;
break;
}
else if(s[i]=='4'){
if(n2if(n1'4';
n1++;
}
else if(n1==n){
ss[i]='7';
flag=i+1;
break;
}
}
else if(s[i]>'4'&&s[i]<'7'){
if(n2'7';
flag=i+1;
break;
}
else if(n2==n){
f=1;
break;
}
}
else if(s[i]=='7'){
if(n2'7';
n2++;
}
else{
f=1;
break;
}
}
else if(s[i]>'7'){
f=1;
break;
}
}
if(f){
ss[pos]='7';
flag=pos+1;
}
int num1=0,num2=0;
for(int i=0;iprintf("%c",ss[i]);
if(ss[i]=='4') num1++;
if(ss[i]=='7') num2++;
}
for(int i=0;iprintf("4");
for(int i=0;iprintf("7");
printf("\n");
}
}
}
return 0;
}
官方题解:
直接暴力显然TLE,考虑按位DFS
每一位只可能是4或7
所以根据这个来DFS即可,时间复杂度O(T∗2log10n)O(T*2^{log_{10}n})O(T∗2log10n)
考虑到T特别大,不可能每次都DFS
而经过计算,218=2621442^{18}=262144218=262144,所以全部储存下来
对于每次询问,二分即可
考虑一个边界条件,即当结果爆ll怎么办?
即答案应当为10个4、10个7的时候,显然unsigned long long也不行,那么只能采用特判了。
#include
using namespace std;
typedef long long ll;
int m;
ll a[1000000];
void dfs(int v4,int v7,ll x)
{
if(v4==v7){
if(x) a[m++]=x;
}
if(v4<9) dfs(v4+1,v7,x*10+4);
if(v7<9) dfs(v4,v7+1,x*10+7);
}
const string s="44444444447777777777";
int main()
{
m=0;dfs(0,0,0);
sort(a,a+m);
int T;scanf("%d",&T);
while(T--){
ll n;scanf("%lld",&n);
int p=lower_bound(a,a+m,n)-a;
if(pprintf("%lld\n",a[p]);
else cout<