URAL 1658. Sum of Digits(DP)

题目链接

隔了一年零三个月，重新刷URAL，这题挺麻烦的输出路径。输出路径挺扯的，乱写了写乱改改就A了。。。我本来想用很靠谱，记录每一条路径的，然后输出最小的，结果Tle，然后我使劲水水又过了一组，发现别人的题解。。直接来了一次 就过了。。我乱搞了搞，倒着记录最小的，然后倒着输出，就过了。。。

#include <cstring>
#include <cstdio>
#include <string>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
#define INF 10000000
int dp[901][8101];
int pre[901][8101];
int s[1010];
struct node
{
    int a,b;
};
void spfa()
{
    int i,a,b;
    queue<node> que;
    node temp,u,v;
    for(i = 1; i < 10; i ++)
    {
        temp.a = i;
        temp.b = i*i;
        dp[i][i*i] = 1;
        pre[i][i*i] = i;
        que.push(temp);
    }
    while(!que.empty())
    {
        u = que.front();
        a = u.a;
        b = u.b;
        que.pop();
        if(a > 900) continue;
        if(b > 8100) continue;
        for(i = 1; i < 10; i ++)
        {
            v.a = i+a;
            v.b = i*i+b;
            if(i + a > 900) continue;
            if(i*i + b > 8100) continue;
            if(dp[v.a][v.b] > dp[u.a][u.b] + 1)
            {
                pre[v.a][v.b] = i;
                dp[v.a][v.b] = dp[u.a][u.b] + 1;
                que.push(v);
            }
        }
    }
    return ;
}

int main()
{
    int t,n,m,i,j;
    int top;
    for(i = 1; i <= 900; i ++)
    {
        for(j = 1; j <= 8100; j ++)
        {
            dp[i][j] = INF;
            pre[i][j] = INF;
        }
    }
    spfa();
    scanf("%d",&t);
    top = 0;
    while(t--)
    {
        scanf("%d%d",&n,&m);
        if(n > 900||m > 8100)
        {
            printf("No solution
");
            continue;
        }
        else if(dp[n][m] > 100)
        {
            printf("No solution
");
            continue;
        }
        int top = 0;
        while(n&&m)
        {
            i = pre[n][m];
            s[top++] = i;
            n = n - i;
            m = m - i*i;
        }
        for(i = top-1;i >= 0;i --)
        printf("%d",s[i]);
        printf("
");
    }
    return 0;
}