1. 程式人生 > >貪心演算法:特殊密碼鎖

貪心演算法:特殊密碼鎖

/*
 * 1.cpp
 *
 *  Created on: 2017年8月30日
 *      Author: Administrator
 */


#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
string result,s,lock;
int temp = 0 ,answer = 1e9, n;
void press(int i){
	s[i-1] = s[i-1] == '1'?'0':'1';
	s[i] = s[i] == '1'?'0':'1';
	if(i+1<n)
		s[i+1] = s[i+1] == '1'?'0':'1';
}

int main(){
	cin>>lock>>result;
	s = lock;
	n = lock.size();
	//第一種情況,當前字元不相同時,從下一個字元開始按按鈕
	for( int i = 1; i<n; i++){
		if(s[i-1] != result[i-1]){
			press(i);
			temp ++ ;
		}
	}
	//可能的結果一:
	if(s[n-1] == result[n-1]){
		answer = temp ;
	}

//	第二種情況:如果前兩個按鈕不匹配,按第一個按鈕,其他按鈕不匹配還是按後一個按鈕
	s = lock;
	temp = 1;
	if( s[0] != result[0] || s[1] != result[1]){
		s[0] = s[0] == '1'?'0':'1';
		s[1] = s[1] == '1'?'0':'1';
		for(int i = 1; i< n; i++){
			if(s[i-1] != result[i-1]){
				press(i);
				temp ++;
			}
		}
	}

	//可能的結果二:
		if(s[n-1] == result[n-1]){
			answer = min(answer,temp);
		}

	if(answer == 1e9){
		cout<< "impossible";
	}
	else{
		cout<< answer;
	}

	 return 0 ;
}

思路:看到輸入輸出首先想到的是列舉所有按鈕的狀態,但是n的範圍為30,所以會有2的30次方多種,所以肯定不能枚舉出所有狀態,於是用到了一個貪心策略,從左往右,如果按鈕不匹配就按下一個按鈕,始終讓左面的按鈕是匹配的,如果遍歷到最後一個按鈕不匹配則"impossible",否則輸出最少的按鈕次數。但是容易忽略一個特殊情況,即前兩個按鈕,當前兩個按鈕不匹配時既可以按第一個按鈕也可以按第二個按鈕,所以應當考慮這兩種情況最後哪中情況按的次數少。