URAL 1658 Sum of Digits

URAL 1658

思路：

dp+记录路径

状态：dp[i][j]表示s1为i,s2为j的最小位数

初始状态：dp[0][0]=0

状态转移：dp[i][j]=min(dp[i-k][j-k*k]+1,dp[i][j])(0<=k<10)

在状态转移时用一个数组v[i][j]记录选的k

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))

const int INF=0x3f3f3f3f;
int dp[][];
int v[][];
vector<int>ans;
int main(){
    ios::sync_with_stdio(false);
    cin.tie();
    mem(dp,INF);
    dp[][]=;
    for(int i=;i<=;i++){
        for(int j=;j<=;j++){
            for(int k=;k<=;k++){
                if(i-k>=&&j-k*k>=){
                    if(dp[i-k][j-k*k]+<dp[i][j]){
                        dp[i][j]=dp[i-k][j-k*k]+;
                        v[i][j]=k;
                    }
                }
            }
        }
    }
    int T,s1,s2;
    cin>>T;
    while(T--){
        cin>>s1>>s2;
        if(s1>||s2>||dp[s1][s2]>){
            cout<<"No solution"<<endl;
            continue;
        }
        ans.clear();
        while(true){
            if(s1==&&s2==)break;
            ans.pb(v[s1][s2]);
            int t=v[s1][s2];
            s1-=t;
            s2-=t*t;
        }
        sort(ans.begin(),ans.end());
        for(int i=;i<ans.size();i++)cout<<ans[i];
        cout<<endl;
    }
    return ;
}