洛谷2657 windy数(数位DP)

传送门

【题目分析】

数位DP经典题了。

考虑直接统计R内的windy数和L-1内的windy数,两者相减即为L~R之间的windy数。

考虑DP,记录当前位以及上一位所填的数,当前是否前面为前导零,以及前面一段是否与原数相同。

然后按题意逐位计算即可。

然后就会惊讶的发现TLE了。。。。。。

所以考虑优化:如果当前没有受任何限制,那么可以直接记录下当前位置的值,记忆化搜索。

【代码~】

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MAXN=15;

LL l,r;
int num[MAXN],cnt;
int dp[MAXN][MAXN];

LL Read(){
	LL i=0,f=1;
	char c;
	for(c=getchar();(c>'9'||c<'0')&&c!='-';c=getchar());
	if(c=='-')
	  f=-1,c=getchar();
	for(;c>='0'&&c<='9';c=getchar())
	  i=(i<<3)+(i<<1)+c-'0';
	return i*f;
}

int DP(int pos,int last,int iszero,int limit){
	if(pos==0)
	  return !iszero;
	if(!iszero&&!limit&&dp[pos][last]!=-1)
	  return dp[pos][last];
	int sx=limit?num[pos]:9,ret=0;
	for(int i=0;i<=sx;++i){
		int delta=abs(i-last);
		if(!iszero){
			if(delta>=2){
				ret+=DP(pos-1,i,iszero,limit&&i==num[pos]);
			}
		}
		else{
			ret+=DP(pos-1,i,i==0,limit&&i==num[pos]);
		}
	}
	if(!iszero&&!limit)
	  return dp[pos][last]=ret;
	return ret;
}

int solve(int x){
	cnt=0;
	while(x){
		num[++cnt]=x%10;
		x/=10;
	}
	return DP(cnt,0,1,1);
}

int main(){
	memset(dp,-1,sizeof(dp));
	l=Read(),r=Read();
	cout<<solve(r)-solve(l-1);
	return 0;
}

 

posted @ 2018-11-07 21:17  Ishtar~  阅读(148)  评论(0编辑  收藏  举报