1. 程式人生 > >hdu3709(求區間內平衡數的個數)數位dp

hdu3709(求區間內平衡數的個數)數位dp

題意:題中平衡數的定義: 以一個位置作為平衡軸,然後左右其他數字本身大小作為重量,到平衡軸的距離作為全職,實現左右平衡(即槓桿原理平衡)。然後為區間[x,y]內平衡數的個數。 (0 ≤ x ≤ y ≤ 1018)

解法:數位dp。如果一個數的平衡數,那麼它的平衡軸位置是確定的。原來一直嘗試數位dp在dfs時候列舉平衡軸的位置,後來才意識到可以提前列舉平衡軸位置,然後再dfs,這樣比較好寫。dp[mid][pre][wei];表示對稱軸是mid,計算第pre個位置以後需要力矩大小wei的數的個數。

     ps:這個題關鍵是跟著適牛學了點東西,首先原來自己寫了一個now的無用的東西一維記錄前一個數字是啥(受前邊題目影響,習慣性思維),但是後來發現明顯沒用。雖然提交A掉了,但是我又嘗試去掉那一維試了一下,發現wa了,半天也沒發現錯哪了,後來適牛提醒說有可能是陣列越界了。最後才發現真相:由於原來多了一維,為陣列越界提供了可靠的空間,這也是為什去掉之後才會wa的原因。開大了點wei的那一維就又AC了。還從適牛那裡學到:由於mid確定後,dp是不變的,所以只需要memset一次就夠了,加上wei>=(len*(len+1)/2*9)+2的剪枝後,做到了0

ms。

程式碼:

/******************************************************
* author:xiefubao
*******************************************************/
#pragma comment(linker, "/STACK:102400000,102400000")
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <queue>
#include <vector>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <stack>
#include <string.h>
//freopen ("in.txt" , "r" , stdin);
using namespace std;

#define eps 1e-8
const double pi=acos(-1.0);
typedef long long LL;
const int Max=10100;
const int INF=1000000007;
LL dp[20][20][410];
int num[20];
LL l,r;
int len=0;
LL dfs(int mid,int pre,int wei,bool limit)
{
    if(wei<0||wei>=(len*(len+1)/2*9)+2)
        return 0;
    if(pre==0)
        return wei==0;
    if(!limit&&dp[mid][pre][wei]!=-1)
        return dp[mid][pre][wei];
    int en=limit ? num[pre-1]:9;
    LL ans=0;
    for(int i=0; i<=en; i++)
    {
        if(pre-1==mid&&i==0&&wei==0) continue;
        ans+=dfs(mid,pre-1,wei+(pre-1-mid)*i,limit&&(i==en));
    }
    return (!limit) ? dp[mid][pre][wei]=ans : ans;
}
LL solve(LL n)
{
    if(n<0)
        return 0;
    int p=0;
    while(n)
    {
        num[p++]=n%10;
        n/=10;
    }
    len=p/2;
    LL ans=0;
    for(int mid=p-1; mid>=0; mid--)
    {
        for(int i=0; i<=num[p-1]; i++)
        {
          if(i==0&&mid==p-1) continue;
            ans+=dfs(mid,p-1,(p-1-mid)*i,i==num[p-1]);
        }
    }
    return ans+1;
}
int main()
{
    int t;
    cin>>t;
    memset(dp,-1,sizeof dp);
    while(t--)
    {
        scanf("%I64d%I64d",&l,&r);
        cout<<solve(r)-solve(l-1)<<'\n';
    }
    return 0;
}