DFS+剪枝 POJ2362的强化版,重点在于剪枝 令InitLen为所求的最短原始棒长,maxlen为给定的棒子堆中最长的棒子,sumlen为这堆棒子的长度之和,那么InitLen必定在范围[maxlen,sumlen]中 根据棒子的灵活度(棒子越长,灵活度越低) DFS前先对所有棒子降序排序 剪枝: 1、 由于所有原始棒子等长,那么必有sumlen%Initlen==0; 2、 若能在[maxlen,sumlen-InitLen]找到最短的InitLen,该InitLen必也是[maxlen,sumlen]的最短;若不能在[maxlen,sumlen-InitLen]找到最短的InitLen,则必有InitLen=sumlen; 3、 由于所有棒子已降序排序,在DFS时,若某根棒子不合适,则跳过其后面所有与它等长的棒子; 4、 最重要的剪枝:对于某个目标InitLen,在每次构建新的长度为InitLen的原始棒时,检查新棒的第一根stick[i],若在搜索完所有stick[]后都无法组合,则说明stick[i]无法在当前组合方式下组合,不用往下搜索(往下搜索会令stick[i]被舍弃),直接返回上一层
#include<iostream> #include<algorithm> using namespace std; int cmp( const void* a, const void* b ) { return *( int* )b - *( int* )a; } int n; bool dfs( int* stick, bool* visit, int len, int InitLen, int s, int num ) { if ( num == n ) return true; int sample = -1; for( int i = s; i<n; i++ ) { if( visit [i] || stick[i] == sample ) continue; visit[i] = true; if( len + stick[i] < InitLen ) { if( dfs( stick, visit, len+stick[i], InitLen, i, num+1)) return true; else sample = stick[i]; } else if( len + stick[i] == InitLen ) { if( dfs( stick, visit, 0, InitLen, 0, num+1 )) return true; else sample = stick[i]; } visit[i] = false; if( len == 0 ) break; } return false; } int main() { while( cin >> n && n ) { int* stick = new int[n]; bool* visit = new bool[n]; int sumlen = 0; int i; for( i = 0; i<n; i++ ) { cin >> stick[i]; sumlen+=stick[i]; visit[i] = false; } qsort( stick, n, sizeof(stick), cmp ); int maxlen = stick[0]; bool flag = false; for ( int InitLen = maxlen;InitLen <= sumlen - InitLen; InitLen++) { if( !( sumlen%InitLen) && dfs( stick, visit, 0, InitLen, 0, 0 )) { cout<< InitLen <<endl; flag = true; break; } } if( !flag ) cout<< sumlen << endl; delete stick; delete visit; } return 0; }