想写这个题很久了……现场赛的时候明知是DP都没有任何思路,如今再来做这题还是很纠结。写到晚上4点还不多,今天中午发现是自己的偏移量算的有问题……。借70学长的代码看了一下,感觉状态是一样的,最后A掉了这个题。很有成就感,好久没有为1个题这么费劲了
题目要求求出[x, y] 范围内的平衡数,平衡数定义为:以数中某个位为轴心,两边的数的偏移量为矩,数位权重,使得整个数平衡。
首先要明确的是一个平衡数只能有一个轴(含前导零的0除外,要特殊处理)。
那么DP的状态很明显是dp[i][j][k]处理到第i位 以j为轴 左边比右边大k的个数。
然后看一下这个状态如何转移,以12300为例
首先从高位开始枚举,维护一个无公共前缀的,这个方法同一般的位DP一样,详情可以参考我之前的解题报告。
然后说一下这个题轴的问题,以上数据为例。当我们枚举到1YX1X2X3(Y是当前数字)
1.可以有轴为 0 to i - 1 前i - 1位为固定的数 第i位(Y)是不大于digit[Y]的数,i + 1以后为0 - 9。轴是i + 1 以后的任何一个,这种情况相对比较简单。
2.还有一种可能就是枚举到1YX1X2X3是,我们让Y等于digit[Y]且为轴,再从i + 1开始重新维护一个小于原来数字后 i + 1位的前缀,且计数。这是一个非常容易忘记的情况。例如 12100这个数就有可能被忽略了;(我比较脑惨想了好久才反映过来)
我们设 DP3(len, piv, pre)为长度为i 以piv位为轴, 高pre位不变的的平衡数的个数。
DP3里的转移过程就比较简单了,写过就是一个类似背包的过程。
最后0的处理:0其实总共计算了len次,所以 - len + 1就可以了
整体的代码如下:
#include <iostream>
#include <cstring>
#include <cstdio>
usingnamespace std;
#define DEBUG(x) cout << #x << " " << x << endl;
constint LEN =20;
constint STATUS =406;
int digit[LEN];
int status(int pre, int piv, int pos, int v)
{
return pre + (pos - piv) * v;
}
longlong DP3(int len, int piv, int pre)
{
longlong dp[LEN][STATUS] = {0};
dp[len][0] =1;
for(int j = len -1; j >=0; j--)
{
int maxn =10, minn =0;
if(j == pre) maxn = digit[j +1];
if(j > pre) minn = digit[j +1], maxn = digit[j +1] +1;
for(int s =0; s < STATUS; s++)
{
if(!dp[j +1][s]) continue;
for(int k = minn; k < maxn; k++)
{
int now = status(s, piv, j, k);
if(now <0|| now >= STATUS) continue;
dp[j][now] += dp[j +1][s];
}
}
}
return dp[0][0];
}
longlong DP(longlong n)
{
if(n ==0) return0;
int len =0;
while(n !=0)
{
digit[++len] = n %10;
n /=10;
}
longlong tmp =0;
longlong ans =0;
for(int i = len; i >0; i--)
{
for(int k = i -1; k >=0; k--)
ans += DP3(len, k, i -1);
for(int k = i -1;k >0;k--)
tmp += DP3(len, i -1, k -1);
}
return tmp + ans - len +1;
}
int main()
{
longlong a, b;
int cases;
cin >> cases;
while(cases--)
{
cin >> a >> b;
cout << DP(b +1) - DP(a) << endl;
}
return0;
}
最后无比仰慕70教主的代码,我的代码由于要多次整体memset,倒是速度非常慢。
70的状态时dp[i][j][k][2]最后一维表示是否达前i位是否达到上限,所以更加直观明显。效率也非常优秀!