LightOJ 1061 N Queen Again（记忆化搜索）

题目链接：http://www.lightoj.com/volume_showproblem.php?problem=1061

题意：给定8*8的棋盘上8个皇后。挪动最少的步数使得不互相攻击?

思路：首先预处理求出所有的合法放法（就是这里一开始想了好久没想出来）。然后对于每个测试数据，枚举每一种合法放法。。。枚举的时候记忆化搜索和直接BFS都行，前一种快点。。

 

View Code

  1 //记忆化搜索

  2 #include <iostream>

  3 #include <cstdio>

  4 #include <cstring>

  5 #define min(x,y) ((x)<(y)?(x):(y))

  6 using namespace std;

  7  

  8 struct node

  9 {

 10     int x[10],y[10];

 11 };

 12  

 13 node ans[1000],p;

 14 int sum;

 15 int C,num=0;

 16 char s[10][10];

 17  

 18 int abs(int x)

 19 {

 20     return x>0?x:-x;

 21 }

 22  

 23 inline int OK(int x,int y,node a,int n)

 24 {

 25     int i;

 26     for(i=1;i<=n;i++) if(a.y[i]==y||abs(a.x[i]-x)==abs(a.y[i]-y))

 27        return 0;

 28     return 1;

 29 }

 30  

 31 void DFS(int dep,node &a)

 32 {

 33     if(dep==9)

 34     {

 35         ans[++sum]=a;

 36         return;

 37     }

 38     int i;

 39     for(i=1;i<=8;i++) if(OK(dep,i,a,dep-1))

 40     {

 41         a.x[dep]=dep;

 42         a.y[dep]=i;

 43         DFS(dep+1,a);

 44     }

 45 }

 46  

 47 void init()

 48 {

 49     sum=0;

 50     node a;

 51     DFS(1,a);

 52 }

 53  

 54 const int INF=1000000000;

 55 int f[10][300];

 56  

 57 inline int get(node a,int t1,node b,int t2)

 58 {

 59     int x1=a.x[t1],y1=a.y[t1];

 60     int x2=b.x[t2],y2=b.y[t2];

 61     if(x1==x2&&y1==y2) return 0;

 62     if(x1==x2||y1==y2||abs(x1-x2)==abs(y1-y2)) return 1;

 63     return 2;

 64 }

 65  

 66 int DFS1(int u,int st,node a)

 67 {

 68     if(u==8) return 0;

 69     if(f[u][st]!=-1) return f[u][st];

 70     int i,t=INF;

 71     for(i=0;i<8;i++) if(0==(st&(1<<i)))

 72         t=min(t,get(p,u+1,a,i+1)+DFS1(u+1,st|(1<<i),a));

 73     return f[u][st]=t;

 74 }

 75  

 76 int deal()

 77 {

 78     int temp=INF,i;

 79     for(i=1;i<=sum;i++)

 80     {

 81         memset(f,-1,sizeof(f));

 82         temp=min(temp,DFS1(0,0,ans[i]));

 83     }

 84     return temp;

 85 }

 86  

 87 int main()

 88 {

 89     init();

 90     for(scanf("%d",&C);C--;)

 91     {

 92         int i,j,k=0;

 93         for(i=1;i<=8;i++)

 94         {

 95             scanf("%s",s[i]+1);

 96             for(j=1;j<=8;j++) if(s[i][j]=='q')

 97             {

 98                 p.x[++k]=i;

 99                 p.y[k]=j;

100             }

101         }

102         printf("Case %d: %d\n",++num,deal());

103     }

104     return 0;

105 }

106 

107 //直接BFS

108 #include <iostream>

109 #include <cstdio>

110 #include <cstring>

111 #include <queue>

112 #define min(x,y) ((x)<(y)?(x):(y))

113 using namespace std;

114 

115 struct node

116 {

117     int x[10],y[10];

118 };

119 

120 node ans[1000],p;

121 int sum;

122 int C,num=0;

123 char s[10][10];

124 

125 int abs(int x)

126 {

127     return x>0?x:-x;

128 }

129 

130 inline int OK(int x,int y,node a,int n)

131 {

132     int i;

133     for(i=1;i<=n;i++) if(a.y[i]==y||abs(a.x[i]-x)==abs(a.y[i]-y))

134         return 0;

135     return 1;

136 }

137 

138 void DFS(int dep,node &a)

139 {

140     if(dep==9)

141     {

142         ans[++sum]=a;

143         return;

144     }

145     int i;

146     for(i=1;i<=8;i++) if(OK(dep,i,a,dep-1))

147     {

148         a.x[dep]=dep;

149         a.y[dep]=i;

150         DFS(dep+1,a);

151     }

152 }

153 

154 void init()

155 {

156     sum=0;

157     node a;

158     DFS(1,a);

159 }

160 

161 

162 

163 inline int get(node a,int t1,node b,int t2)

164 {

165     int x1=a.x[t1],y1=a.y[t1];

166     int x2=b.x[t2],y2=b.y[t2];

167     if(x1==x2&&y1==y2) return 0;

168     if(x1==x2||y1==y2||abs(x1-x2)==abs(y1-y2)) return 1;

169     return 2;

170 }

171 

172 const int INF=1000000000;

173 int f[66000],visit[66000];

174 

175 int cal(node a,node b)

176 {

177     int k,aa,bb,k0,aa0,bb0,i,j,cost;

178     queue<int> Q;

179     memset(f,-1,sizeof(f));

180     memset(visit,0,sizeof(visit));

181     f[255]=0;

182     visit[255]=1;

183     Q.push(255);

184     while(!Q.empty())

185     {

186         k=Q.front();

187         Q.pop();

188         visit[k]=0;

189         aa=k>>8;

190         bb=k&255;

191         for(i=0;i<8;i++) if(bb&(1<<i))

192         {

193             for(j=0;j<8;j++) if(0==(aa&(1<<j)))

194             {

195                 aa0=aa^(1<<j);

196                 bb0=bb^(1<<i);

197                 k0=(aa0<<8)+bb0;

198                 cost=get(b,i+1,a,j+1);

199                 if(f[k0]==-1||f[k0]>f[k]+cost)

200                 {

201                     f[k0]=f[k]+cost;

202                     if(!visit[k0])

203                     {

204                         Q.push(k0);

205                         visit[k0]=1;

206                     }

207                 }

208             }

209             break;

210         }

211     }

212     return f[255<<8];

213 }

214 

215 int deal()

216 {

217     int temp=INF,i;

218     for(i=1;i<=sum;i++)

219     {

220         memset(f,-1,sizeof(f));

221         temp=min(temp,cal(ans[i],p));

222     }

223     return temp;

224 }

225 

226 int main()

227 {

228     init();

229     for(scanf("%d",&C);C--;)

230     {

231         int i,j,k=0;

232         for(i=1;i<=8;i++)

233         {

234             scanf("%s",s[i]+1);

235             for(j=1;j<=8;j++) if(s[i][j]=='q')

236             {

237                 p.x[++k]=i;

238                 p.y[k]=j;

239             }

240         }

241         printf("Case %d: %d\n",++num,deal());

242     }

243     return 0;

244 }