hdu5676 ztr loves lucky numbers

题意:找到第一个比n大的幸运数,幸运数的定义是只由4或7组成,且4的个数等于7。
开始我的做法是暴力,按照每位贪心的找,坑点和要考虑的地方很多,写的很麻烦。

#include<bits/stdc++.h>
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;i<len;i++){
        if(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;i<n;i++) printf("4");
            for(int i=0;i<n;i++) printf("7");
            printf("\n");
        }
        else{
            int n=len/2;
            for(int i=0;i<n;i++) ss[i]='7';
            for(int i=0;i<n;i++) ss[i+n]='4';
            ss[len]=0;
            if(strcmp(ss,s)<0){
                n++;
                for(int i=0;i<n;i++) printf("4");
                for(int i=0;i<n;i++) printf("7");
                printf("\n");
            }
            else{
                int pos=0;
                int flag=0;
                int f=0;
                int n1=0,n2=0;
                for(int i=0;i<len;i++){
                    if(s[i]<'4'){
                        if(n1<n) ss[i]='4';
                        else if(n1==n) ss[i]='7';
                        flag=i+1;
                        break;
                    }
                    else if(s[i]=='4'){
                        if(n2<n) pos=i;
                        if(n1<n){
                            ss[i]='4';
                            n1++;
                        }
                        else if(n1==n){
                            ss[i]='7';
                            flag=i+1;
                            break;
                        }
                    }
                    else if(s[i]>'4'&&s[i]<'7'){
                        if(n2<n){
                            ss[i]='7';
                            flag=i+1;
                            break;
                        }
                        else if(n2==n){
                            f=1;
                            break;
                        }
                    }
                    else if(s[i]=='7'){
                        if(n2<n){
                            ss[i]='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;i<flag;i++){
                    printf("%c",ss[i]);
                    if(ss[i]=='4') num1++;
                    if(ss[i]=='7') num2++;
                }
                for(int i=0;i<n-num1;i++) printf("4");
                for(int i=0;i<n-num2;i++) printf("7");
                printf("\n");
            }
        }
    }
    return 0;
}

官方题解:
直接暴力显然TLE,考虑按位DFS
每一位只可能是4或7
所以根据这个来DFS即可,时间复杂度O(T∗2log10n)O(T*2^{log_{10}n})O(T∗2​log​10​​n​​)
考虑到T特别大,不可能每次都DFS
而经过计算,218=2621442^{18}=2621442​18​​=262144,所以全部储存下来
对于每次询问,二分即可
考虑一个边界条件,即当结果爆ll怎么办?
即答案应当为10个4、10个7的时候,显然unsigned long long也不行,那么只能采用特判了。

#include<bits/stdc++.h>
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(p<m) printf("%lld\n",a[p]);
        else cout<<s<<endl;
    }
}

你可能感兴趣的:(hdu5676 ztr loves lucky numbers)