【URAL 1658】1658. Sum of Digits(DP)
题目大意:
定义s1为n的每位数字的加和。
s2为n的每位数字的平方和。
给出s1,s2,输出最小的n。如果不存在或者长度超过100,输出No solution。
dp[i][j]表示加和为i,平方和为j的最短长度。
预处理出来所有的加和和平方和的组合,然后存一下每个最小转移添加的数字。
因为长度超过100输出No solution。所以s1 > 900的直接pass。dp数组的第一维就可以变成901。
然后对于每个输入递归输出。
需要注意的是转移的时候如果有多个最优转移,存数字较小的。
for(int k = 1; k <= 9 && i-k >= 0 && j-k*k >= 0; ++k)
{
if(dp[i][j] > dp[i-k][j-k*k]+1)//大于号而不是大于等于!
{
dp[i][j] = dp[i-k][j-k*k]+1;
ned[i][j] = k;
}
}
因为对于相同长度的n,高位越小,n越小,所以递归的时候从高位开始填数,优先添较小的数。
代码如下:
#include <iostream>
#include <cmath>
#include <vector>
#include <cstdlib>
#include <cstdio>
#include <climits>
#include <ctime>
#include <cstring>
#include <queue>
#include <stack>
#include <list>
#include <algorithm>
#include <map>
#include <set>
#define LL long long
#define Pr pair<int,int>
#define fread(ch) freopen(ch,"r",stdin)
#define fwrite(ch) freopen(ch,"w",stdout)
using namespace std;
const int INF = 0x3f3f3f3f;
const int mod = 1e9+7;
const double eps = 1e-8;
const int maxn = 112345;
int dp[999][11234];
int ned[999][11234];
int ans[233];
int tp;
void init()
{
memset(dp,INF,sizeof(dp));
dp[0][0] = 0;
for(int i = 1; i <= 900; ++i)
{
for(int j = i; j <= 10000 && j <= i*i; ++j)
{
for(int k = 1; k <= 9 && i-k >= 0 && j-k*k >= 0; ++k)
{
if(dp[i][j] > dp[i-k][j-k*k]+1)
{
dp[i][j] = dp[i-k][j-k*k]+1;
ned[i][j] = k;
}
}
}
}
}
bool solve(int a,int b)
{
if(a > 900 || dp[a][b] > 100) return false;
if(dp[a][b] == 0) return true;
if(!ned[a][b]) return false;
ans[tp++] = ned[a][b];
if(solve(a-ned[a][b],b-ned[a][b]*ned[a][b])) return true;
}
int main()
{
//fread("");
//fwrite("");
init();
int t,a,b;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&a,&b);
tp = 0;
if(!solve(a,b)) puts("No solution");
else
{
for(int i = 0; i < tp; ++i) printf("%d",ans[i]);
puts("");
}
}
return 0;
}