【HDOJ6351】Beautiful Now（贪心，搜索）

题意：给定一个数字n，最多可以交换其两个数位k次，求交换后的最大值与最小值，最小值不能有前导0

n,k<=1e9

思路：

当k>=n的位数时只需要无脑排序

k<n时有一个显然的贪心是从高位到低位，如果当前位置不是可取的最值的话就从比现在低的位置挑一个最值换上来

问题在于可能有多个低位上的数字相同，而且这个问题没有正确的贪心策略

所以只能记录下它们的位置，依次搜索

搜索最小值时第一位不能是0

#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<vector>
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef vector<int> VI;
#define fi first
#define se second 
#define MP make_pair
#define N   11000000
#define MOD 1000000007
#define eps 1e-8 
#define pi acos(-1)

ll f[10],mx,mn;
int a[10],b[10],K,len;

void dfs1(ll n,int s,int k)
{
//printf("%lld %d %d\n",n,s,k); 
    mn=min(mn,n);
    mx=max(mx,n);
    if(s==len) return;
    if(k==K+1) return;
    int flag=1;
    for(int i=s+2;i<=len;i++)
     if(b[i]<b[s+1]) {flag=0;break;}
    if(flag)
    {
        dfs1(n,s+1,k);
        return;
    }
    int c[9];
    int now=b[s+1];
    int m=0;
    for(int i=s+2;i<=len;i++)
    { 
         if(b[i]<now)
         {
             m=1;
             c[1]=i;
             now=b[i];
             continue;
        }
        if(b[i]==now) c[++m]=i;
    }
    //printf("%d %d\n",m,c[m]); 
    if(k==1&&s==0&&m>0&&b[c[m]]==0)
    {
        m=0; now=b[1];
        for(int i=2;i<=len;i++)
        {
            if(b[i]==0) continue;
            if(b[i]<now)
             {
                 m=1;
                 c[1]=i;
                 now=b[i];
                 continue;
            }
            if(b[i]==now) c[++m]=i;
        }
    }
    for(int i=1;i<=m;i++)
    {
        int t=c[i];
        ll tmp=n+f[len-s-1]*(b[t]-b[s+1])+f[len-t]*(b[s+1]-b[t]);
        swap(b[s+1],b[t]);
        dfs1(tmp,s+1,k+1); 
        swap(b[s+1],b[t]);
    }
    dfs1(n,s+1,k);
}
        
void dfs2(ll n,int s,int k)
{
    //printf("%lld %d %d\n",n,s,k); 
    mn=min(mn,n);
    mx=max(mx,n);
    if(s==len) return;
    if(k==K+1) return;
    int flag=1;
    for(int i=s+2;i<=len;i++)
     if(b[i]>b[s+1]) {flag=0;break;}
    if(flag)
    {
        dfs2(n,s+1,k);
        return;
    }
    int c[9];
    int now=b[s+1];
    int m=0;
    for(int i=s+2;i<=len;i++)
    { 
         if(b[i]>now)
         {
             m=1;
             c[1]=i;
             now=b[i];
             continue;
        }
        if(b[i]==now) c[++m]=i;
    }
    for(int i=1;i<=m;i++)
    {
        int t=c[i];
        ll tmp=n+f[len-s-1]*(b[t]-b[s+1])+f[len-t]*(b[s+1]-b[t]);
        swap(b[s+1],b[t]);
        dfs2(tmp,s+1,k+1); 
        swap(b[s+1],b[t]);
    }
    dfs2(n,s+1,k); 
}
                
    
int read()
{ 
   int v=0,f=1;
   char c=getchar();
   while(c<48||57<c) {if(c=='-') f=-1; c=getchar();}
   while(48<=c&&c<=57) v=(v<<3)+v+v+c-48,c=getchar();
   return v*f;
}

void swap(int &x,int &y)
{
    int t=x;x=y;y=t;
}

int main()
{
    freopen("1.in","r",stdin);
    freopen("1.out","w",stdout);
    f[0]=1;
    for(int i=1;i<=9;i++) f[i]=f[i-1]*10;
    int cas;
    scanf("%d",&cas);
    while(cas--)
    {
        int n;
        scanf("%d%d",&n,&K);
        len=0;
        int t=n;
        while(t)
        {
            b[++len]=t%10;
            t/=10;
        }
        for(int i=1;i<=len/2;i++) swap(b[i],b[len-i+1]);
        if(K>=len)
        {
            for(int i=0;i<=9;i++) a[i]=0;
            int t=n;
            while(t)
            {
                a[t%10]++;
                t/=10;
            }
            for(int i=1;i<=9;i++)
             if(a[i]){a[i]--; printf("%d",i); break;}
            for(int i=0;i<=9;i++)
             for(int j=1;j<=a[i];j++) printf("%d",i);
            printf(" ");
            for(int i=0;i<=9;i++) a[i]=0;
            t=n;
            while(t)
            {
                a[t%10]++;
                t/=10;
            }
            for(int i=9;i>=0;i--)
             for(int j=1;j<=a[i];j++) printf("%d",i);
            printf("\n");
            continue;
        }
             
        mn=mx=n;
       
        dfs1(n,0,1);
        dfs2(n,0,1);
        printf("%d %d\n",mn,mx);
    }
    return 0;
}