题意:给你一些人的屌丝值Di,计算总屌丝值=sum((k-1)*Di),k为第i个人进入的位置。给你一个栈,可以使人的位置改变。
方法一:记忆化搜索
dp[i][j][k]表示区间[i,j]第i个人放在k位置的屌丝值。
搜索所有的状态,记录已经搜索的状态。
代码:
//记忆化搜索 //这里也可以用二维的数组,另外用一个sum数组和。不过三维的实在是太神了 #include<iostream> #include<vector> #include<string> #include<queue> #include<map> #include<cstdio> #include<cstring> #define maxn 105 #define INF 0xfffffff #define min(a,b) a<b?a:b #define max(a,b) a>b?a:b using namespace std; int n; int dp[maxn][maxn][maxn]; int a[maxn]; int solve_dp(int s,int e,int k) { if(s>e) return 0; if(s==e) return a[s]*(k-1); if(dp[s][e][k]!=INF) return dp[s][e][k];//记忆化搜索 int i,j,thisk,nextk,tp,front,rear; //将[s,e]划分为[s,i],[i+1,e],s可以放的位置从s-e //前一个区间s放在最后,s+1放在原来k的位置 for(i=s;i<=e;i++) { thisk=k+(i-s); nextk=thisk+1; tp=a[s]*(thisk-1); front=solve_dp(s+1,i,k); rear=solve_dp(i+1,e,nextk); dp[s][e][k]=min(dp[s][e][k],tp+front+rear); } return dp[s][e][k]; } int main() { int t,i,j,k; scanf("%d",&t); int tt=1; while(t--) { scanf("%d",&n); for(i=1;i<=n;i++) scanf("%d",&a[i]); for(i=0;i<=n;i++) { for(j=0;j<=n;j++) { for(k=0;k<=n;k++) { dp[i][j][k]=INF;//求最小值,所有的状态初始化为INF } } } printf("Case #%d: %d\n",tt++,solve_dp(1,n,1)); } return 0; }直接用for循环,先把小区间计算出来,计算大区间时转化为小区间。
dp[i][j]表示区间[i,j]第i个人放在i位置的屌丝值。
代码:
//三for递推,把当前选择对后续选择的影响给提前计算了 //用sum数组来存和,能够在每次算区间的时候,把第一个看成1 #include<iostream> #include<vector> #include<string> #include<queue> #include<map> #include<cstdio> #include<cstring> #define maxn 105 #define INF 0xfffffff #define min(a,b) a<b?a:b #define max(a,b) a>b?a:b using namespace std; int n; int dp[maxn][maxn]; int a[maxn],sum[maxn]; int solve_dp() { int i,j,len,s,e,k;//dp[i][j]是把i作为第一个位置的值,思路也是把[i,j]区间分成两个 int tp,front,rear; for(len=1;len<n;len++)//可以从小区间开始,大区间转为小区间 { for(i=1;i+len<=n;i++) { s=i,e=i+len; for(k=s;k<=e;k++) { tp=a[s]*(k-s); front=dp[s+1][k]+dp[k+1][e]; rear=(sum[e]-sum[k])*(k-s+1); dp[s][e]=min(dp[s][e],tp+front+rear); } } } return dp[1][n]; } int main() { int t,i,j,k; scanf("%d",&t); int tt=1; while(t--) { scanf("%d",&n); memset(sum,0,sizeof(sum)); for(i=1; i<=n; i++) { scanf("%d",&a[i]); sum[i]=sum[i-1]+a[i]; } for(i=1; i<=n; i++) { for(j=i+1; j<=n; j++) { dp[i][j]=INF;//求最小值,所有的状态初始化为INF } } printf("Case #%d: %d\n",tt++,solve_dp()); } return 0; }