题意:有n个敌人,每个敌人有a[i]个血量,你能用火球射击第2~n-1个人,射击到第i个人,第i个人减少a血量,第i-1,i+1个人减少b血量,问最少射击几次能将所有人血量变为<0;
解:dfs剪枝,考虑题目中所给的数据量较小;从左到右dfs,在射第i个人时,暴力射击次数,最少能将左边人射死,最大能将三个人射死,这样再做一下最优性剪枝1778ms,勉强能过。
AC代码
#include <iostream>
#include <vector>
using namespace std;
vector<int>v,v1;
const int inf=1e9+10;
int minn,n,aa,b;
int a[110];
void dfs(int po,int num)
{
if (num>=minn) // 最优化剪枝
return;
if (po==n)
{
if (a[po]<0)
{
v1=v;
minn=num;
}
return;
}
for (int i=0;i<=max(max(a[po-1]/b+1,a[po]/aa+1),a[po+1]/b+1);i++)// 暴力次数
{
if (a[po-1]-i*b<0)
{
a[po-1]-=i*b;
a[po]-=i*aa;
a[po+1]-=i*b;
for (int j=1;j<=i;j++)
v.push_back(po);
dfs(po+1,num+i);
for (int j=1;j<=i;j++)
v.pop_back();
a[po-1]+=i*b;
a[po]+=i*aa;
a[po+1]+=i*b;
}
}
}
int main()
{
minn=inf;
cin>>n>>aa>>b;
for (int i=1;i<=n;i++)
cin>>a[i];
dfs(2,0);
cout<<minn<<endl;
for (int i=0;i<v1.size();i++)
cout<<v1[i]<<" ";
cout<<endl;
return 0;
}