工兵排雷
题目描述
某工兵发现了一排地雷,这些地雷并不是连续排列的。每次引爆一个地雷,相邻的地雷均会被引爆。对于不连续的地雷排列,工兵可以选择一个个引爆,也可以把炸药埋在两个地雷之间的所有位置,然后一次性引爆地雷。引爆一个地雷的花费是a元,埋一个位置炸药的花费是b元。问工兵引爆这一排地雷的最小花费是多少?
输入格式
第一行一个正整数T,表示有T个测试例子(1<=T<=100)。
后面T行,每行输入正整数a,正整数b和字符串s(1<=a, b<= 100, |s| < 100), s为只含有’0’或’1’的字符串,'0’表示当前位置没有地雷,'1’表示当前位置有地雷,s的长度为100以内。例如’1001001’表示第1,4,7位置上有地雷,其余位置没有地雷。
输出格式
对于每个测试例子,输出一个引爆所有地雷的最小花费。
样例输入
3
1 1 01000010
5 1 01101110
2 1 10010010
样例输出
2
6
6
思路:
1.首先记录地雷的起始位置和结束位置,特判一下是否有雷了。
2.接着从起始位置到结束位置,记录0的个数,之后用来判断埋雷的地方,用来计算花费,存在vector容器中。
3.计算所有都是单个引爆的花费,即容器的(size+1)*a,因为两个地雷中只有一段没埋雷的地方,所以总共的雷是容器容量+1;
4.开始模拟,遍历容器,少引爆一个雷,用第二种方法埋雷在引爆的方法,判断两者谁小就用谁的方法。
#include<bits/stdc++.h>
using namespace std;
#define NN 10000
main(){
int t;
cin>>t;
while(t--){
int a, b;
cin >> a >> b;
string str;
cin >> str;
int n = str.length();
int l = -1, r = n;
for(int i = 0; i < n; i++){
if(str[i] == '1'){
l = i;
break;
}
}
for(int i = n - 1; i >= 0; i--){
if(str[i] == '1'){
r = i;
break;
}
}
if(l == -1 || r == n){
cout << 0 << "\n";
continue;
}
int cnt = 0;
vector<int> v;
for(int i = l; i <= r; i++){
if(str[i] == '0'){
cnt ++;
}else{
v.push_back(cnt);
cnt = 0;
}
}
v.push_back(cnt);
sort(v.begin(), v.end());
int ans = (v.size() + 1) * a;
int res = ans;
for(auto i : v){
ans -= a;
ans += i * b;
res = min(res, ans);
}
cout << res << "\n";
}
}