貪心演算法:特殊密碼鎖
阿新 • • 發佈:2019-01-25
/* * 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",否則輸出最少的按鈕次數。但是容易忽略一個特殊情況,即前兩個按鈕,當前兩個按鈕不匹配時既可以按第一個按鈕也可以按第二個按鈕,所以應當考慮這兩種情況最後哪中情況按的次數少。