第1次实验——NPC问题(回溯算法、聚类分析)

queens.h

[cpp]  view plain copy

1. #include <vector>  
2. #include <utility>  
3. #include <iterator>  
4.   
5. using namespace std;  
6.   
7. typedef pair<int,int> xy;  
8.   
9. class Queens  
10. {  
11. private:  
12.     int QueensNum;  
13.     vector<xy>    inQueens;  
14.     vector<int> Avai;  
15.     vector<vector<int>> Solution;  
16.   
17.     bool isDiagonal(int row,int col,int srcrow,int srccol);  
18.     bool isBan(int row,int col);  
19.     bool SolveProblem(int row,int col);  
20.   
21. public:  
22.   
23.     Queens(int n)  
24.     {  
25.         QueensNum=n;  
26.         for(int i=0;i<n;i++)  
27.             Avai.push_back(i);  
28.     }  
29.   
30.     void StartProblem();  
31.     void ShowResult();  
32.   
33. };  
34.   
35. void Queens::ShowResult()  
36. {  
37.     for(int i=0;i<Solution.size();i++){  
38.         copy(Solution[i].begin(),Solution[i].end(),ostream_iterator<int>(cout," "));  
39.         cout<<endl;  
40.     }  
41.     cout<<Solution.size()<<endl;  
42. }  
43.   
44. void Queens::StartProblem()  
45. {  
46.     for(int i=0;i<QueensNum;i++)  
47.         SolveProblem(0,i);  
48. }  
49.   
50. bool Queens::SolveProblem(int row,int col)  
51. {  
52.     if( isBan(row,col)==true )  return false;  
53.     inQueens.push_back(xy(row,col));  
54.     if(inQueens.size()==QueensNum)    
55.     {  
56.         vector<int> result;  
57.         for(int i=0;i<inQueens.size();i++)  
58.             result.push_back(inQueens[i].second);  
59.         Solution.push_back(result);  
60.         inQueens.pop_back();    //找到一个解,回溯  
61.         return true;  
62.     }  
63.     row++;  
64.     for(int i=0;i<QueensNum;i++)  
65.         SolveProblem(row,i);  
66.     inQueens.pop_back();        //往后没有解,也回溯  
67. }  
68.   
69. bool Queens::isDiagonal(int row,int col,int srcrow,int srccol)  
70. {  
71.     double k=((row-srcrow)*1.0)/(col-srccol);  
72.     if(k==1 || k==-1) return true;  
73.     else    return false;  
74. }  
75.   
76. bool Queens::isBan(int row,int col)  
77. {  
78.     for(int i=0;i<inQueens.size();i++)  
79.     {  
80.         if(inQueens[i].second==col) return true;  
81.         if(isDiagonal(row,col,inQueens[i].first,inQueens[i].second)==true) return true;  
82.         /*if(inQueens[i].second>0 && inQueens[i].second-1==col ) return true; 
83.         if(inQueens[i].second<QueensNum-1 && inQueens[i].second+1==col)  return true;*/  
84.     }  
85.     return false;  
86. }  

queens.cpp

[cpp]  view plain copy

1. #include <iostream>  
2. #include "queens.h"  
3. using namespace std;  
4.   
5. void main()  
6. {  
7.     Queens queens(8);  
8.     queens.StartProblem();  
9.     queens.ShowResult();  
10. }