Timus 1658

令res[i][j][0]表示数字和为i，数字平方和为j的可能性，要么为0要么为1,；res[i][j][1]表示至少有多少位数字。

res[i][j][0]=res[i-k][j-k*k] ? 1 : 0;(1<=k<=9 , 1<=i<=900,1<=j<=8100)

res[i][j][1]=min(res[i][j],res[i-k][j-k*k]+1);

然后根据res[i][j][1]寻找最优解

需要预处理一下，不然可能TLE，之前没有预处理TLE了n次。。。。。。。

#include<iostream>

#include<cstdio>

#include<cstring>

#include<time.h>

using namespace std;

#define MAX 1234567890

int dp[910][8110][2],res[101];

inline int min(int a,int b)

{

    return a < b ? a : b;

}

int work()

{

    int i,j,k;

    memset(dp,0,sizeof(dp));

    dp[0][0][0]=1;

    for(i=1;i<=900;i++)

        for(j=1;j<=8100;j++) dp[i][j][1]=MAX;

    for(k=1;k<10;k++)

        for(i=k;i<=900;i++)

            for(j=k*k;j<=8100;j++)

                if(dp[i-k][j-k*k][0])

                {

                    dp[i][j][0]=1;

                    dp[i][j][1]=min(dp[i-k][j-k*k][1]+1,dp[i][j][1]);

                }

    return 0;

}

int GetRes(int n,int m)

{

    int i,j,k,count=0;

    i=n; j=m;

    while(i>0 && j>0)

    {

        for(k=1;k<10;k++)

            if(i-k>=0 && j-k*k>=0 && dp[i-k][j-k*k][0] && dp[i][j][1]==dp[i-k][j-k*k][1]+1)

            {

                res[count++]=k; i=i-k; j=j-k*k;

                break;

            }

    }

    return count-1;

}

int main()

{

    int i,m,n,N,count;

    work();

    scanf("%d",&N);

    while(N--)

    {

        scanf("%d %d",&n,&m);

        if(n>900 || m>8100)

        {

            printf("No solution\n");

            continue;

        }

        if(dp[n][m][0]==0 || dp[n][m][1]>100)

        {

            printf("No solution\n");

            continue;

        }

        count=GetRes(n,m);

        

        for(i=0;i<=count;i++)printf("%d",res[i]);

        printf("\n");

    }

    return 0;

}