码迷,mamicode.com
首页 > Windows程序 > 详细

[luoguP2657] [SCOI2009]windy数(数位DP)

时间:2018-01-19 14:08:08      阅读:177      评论:0      收藏:0      [点我收藏+]

标签:lan   mat   答案   +=   query   sizeof   amp   str   post   

传送门

 

f[i][j]表示位数为i,第i位为j的windy数的个数

先预处理出f数组。

求的时候先算没有前导0的答案,再算位数和给定的数相同的答案。

 

#include <cmath>
#include <cstdio>
#include <cstring>

using namespace std;

int f[15][10], d[15];

inline void init()
{
	int i, j, k;
	for(i = 0; i <= 9; i++) f[1][i] = 1;
	for(i = 2; i <= 10; i++)
		for(j = 0; j <= 9; j++)
			for(k = 0; k <= 9; k++)
				if(abs(j - k) >= 2) f[i][j] += f[i - 1][k];
}

inline int query(int x)
{
	int i, j, ret = 0;
	memset(d, 0, sizeof(d));
	for(; x; x /= 10) d[++d[0]] = x % 10;
	for(i = 1; i < d[0]; i++)
		for(j = 1; j <= 9; j++) ret += f[i][j];
	for(i = d[0]; i >= 1; i--)
	{
		for(j = 0; j < d[i]; j++)
		{
			if(i == d[0] && !j) continue;
			if(i == d[0]) ret += f[i][j];
			else if(abs(d[i + 1] - j) >= 2) ret += f[i][j];
		}
		if(i != d[0] && abs(d[i + 1] - d[i]) < 2) break;
	}
	return ret;
}

int main()
{
	int l, r;
	init();
	scanf("%d %d", &l, &r);
	printf("%d\n", query(r + 1) - query(l));
	return 0;
}

  

 

[luoguP2657] [SCOI2009]windy数(数位DP)

标签:lan   mat   答案   +=   query   sizeof   amp   str   post   

原文地址:https://www.cnblogs.com/zhenghaotian/p/8316049.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!