藍橋杯 算法提高 合併石子

這是一個動態規劃的題,此題沒有AC,只過了8個測試點,每次相鄰兩堆合成一堆

因爲每次兩堆合成一堆,則是第一堆的最小花費+第二堆的最小花費+兩堆石子之和就是當前合成的最小花費

狀態轉移方程可以寫爲cost[i][j]=min(cost[i][k]+[k+1][j],cost[i][j])+d[j]-d[i]; i==0時要特殊處理一下

#include<iostream>
using namespace std;
const int N=1000+9;
typedef long long LL;
const LL INF=9999999999;
LL cost[N][N];//用於存儲i---j兩點的最小花費
LL d[N];//存儲輸入的數據 
int main()
{
	LL dp(int start,int end);
	int n;
	LL x;
	cin>>n;
	for(int i=0;i<n;i++)
	{
		cin>>d[i];
		if(i>0)d[i]+=d[i-1];//0~i石子之和,方便計算兩堆石子之和 
	}
	for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
			cost[i][j]=INF;
	cout<<dp(0,n-1);
	return 0;
}

LL dp(int start,int end)
{
	if(end-start==0)	return 0;
	if(cost[start][end]<INF)return cost[start][end];
	else
		for(int i=start;i<end;i++)//狀態轉移 兩堆最小的花費加上合併兩堆的花費 
			cost[start][end]=min(cost[start][end],dp(start,i)+dp(i+1,end));
	if(start==0)return cost[start][end]+=d[end];//開頭是0,則需要特殊處理 
	return cost[start][end]+=d[end]-d[start-1];//已經算的要記錄下來,下次使用 
}


發佈了25 篇原創文章 · 獲贊 5 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章