간단한 디지털 dp

4334 단어 dp
디지털 dp 템플릿
hdu 2089
dp[len][flag]를 설정하고 flag=1을 설정하면 앞자리가 6이고 그렇지 않으면 앞자리가 6이 아니다.
#include <stdio.h>
#include <iostream>
#include <map>
#include <set>
#include <list>
#include <stack>
#include <vector>
#include <math.h>
#include <string.h>
#include <queue>
#include <string>
#include <stdlib.h>
#include <algorithm>
#define LL __int64
#define eps 1e-12
#define PI acos(-1.0)
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 50010;

int dig[10];
int dp[10][2];

int dfs(int len, int flag, int up)
{
    if(len == 0)
        return 1;
    if(!up && dp[len][flag] != -1)
        return dp[len][flag];
    int n = up == 1?dig[len]:9;
    int res = 0;
    for(int i = 0; i <= n; i++)
    {
        if(i == 4 || (flag == 1 && i == 2))
            continue;
        res += dfs(len-1,i==6?1:0,up&&(i==n) );
    }
    if(!up)
        dp[len][flag] = res;
    return res;
}

int cal(int num)
{
    int cnt = 0;
    while(num)
    {
        dig[++cnt] = num%10;
        num /= 10;
    }
    memset(dp,-1,sizeof(dp));
    return dfs(cnt,0,1);
}
int main()
{
    int a,b;
    while(~scanf("%d %d",&a,&b))
    {
        if(a == 0 && b == 0)
            break;
        printf("%d
",cal(b) - cal(a-1) ); } return 0; }

hdu 3555
49가 나오면 안 돼요.
#include <stdio.h>
#include <iostream>
#include <map>
#include <set>
#include <list>
#include <stack>
#include <vector>
#include <math.h>
#include <string.h>
#include <queue>
#include <string>
#include <stdlib.h>
#include <algorithm>
#define LL __int64
#define eps 1e-12
#define PI acos(-1.0)
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 4010;

int dig[100];
LL dp[100][4];

LL dfs(int len, int flag, int up)
{
    if(len == 0)
        return flag == 2;
    if(!up && dp[len][flag] != -1)
        return dp[len][flag];

    int n = up ? dig[len] : 9;
    LL res = 0;

    for(int i = 0; i <= n; i++)
    {
        int tflag = flag;
        if(flag == 0 && i == 4)
            tflag = 1;
        else if(flag == 1 && i == 9)
            tflag = 2;
        else if(flag == 1 && i != 4)
            tflag = 0;
        res += dfs(len-1,tflag,up&&(i==n));
    }
    if(!up)
        dp[len][flag] = res;
    return res;
}

LL cal(LL num)
{
    int len = 0;
    while(num)
    {
        dig[++len] = num % 10;
        num /= 10;
    }
    memset(dp,-1,sizeof(dp));
    return dfs(len,0,1);
}

int main()
{
    int test;
    LL n;
    scanf("%d",&test);
    while(test--)
    {
        scanf("%I64d",&n);
        printf("%I64d
",cal(n)); } return 0; }

SCOI2009
두 개의 인접한 수의 차이가 2보다 크도록 요구하는 것은 1위에 대해서는 특수하다.그래서 dfs를 할 때 변수 퍼스트를 추가하면 앞에 모두 0이 있는지 여부를 나타낸다.
#include <stdio.h>
#include <iostream>
#include <map>
#include <set>
#include <list>
#include <stack>
#include <vector>
#include <math.h>
#include <string.h>
#include <queue>
#include <string>
#include <stdlib.h>
#include <algorithm>
#define LL __int64
#define eps 1e-12
#define PI acos(-1.0)
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 4010;

int dig[15];
int dp[15][15]; //dp[i][j]   i       i+1  j


int dfs(int len, int pre, int up, int first)
{
	if(len == 0)
		return first == 0;
	if(!up && dp[len][pre] != -1 && !first)
		return dp[len][pre];
	int n = up ? dig[len] : 9;
	int res = 0;
	for(int i = 0; i <= n; i++)
	{
		if(first) //         , i==0 , i-1      ,    
			res += dfs(len-1,i,up&&i==n,first&&i==0);
		else if(abs(pre-i) >= 2)
			res += dfs(len-1,i,up&&i==n,first);
	}
	if(!up && !first)
		dp[len][pre] = res;
	return res;
}

int cal(int num)
{
	int len = 0;
	while(num)
	{
		dig[++len] = num%10;
		num /= 10;
	}
	memset(dp,-1,sizeof(dp));
	return dfs(len,0,1,1);
}

int main()
{
	int a,b;
	while(~scanf("%d %d",&a,&b))
	{
		printf("%d
",cal(b) - cal(a-1)); } return 0; }

좋은 웹페이지 즐겨찾기