hdu3709(求區間內平衡數的個數)數位dp
阿新 • • 發佈:2019-01-26
題意:題中平衡數的定義: 以一個位置作為平衡軸,然後左右其他數字本身大小作為重量,到平衡軸的距離作為全職,實現左右平衡(即槓桿原理平衡)。然後為區間[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
程式碼:
/****************************************************** * 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; }