USACO-Section1.4 Combination Lock （枚举）

2017-5-30

题目描述

给你两个三位数的排列，求出满足条件的排列总数

解答

暴力枚举法

代码

/*
ID: 18795871
PROG: combo
LANG: C++
*/
#include<iostream>
#include<fstream>
#include<cmath>
using namespace std;

ifstream fin("combo.in");
ofstream fout("combo.out");

int a,b,c,d,e,f,N;

bool res(int m,int n){//求差值
if (abs(m-n)<=2||abs(m-n)>=N-2) return true;
return false;
}

int main()
{
fin>>N;
fin>>a>>b>>c>>d>>e>>f;
int i,j,k,f1,f2,f3,f4,r=0;
for (i=1;i<=N;i++){
f1=0;f2=0;
if (!res(i,a)&&!res(i,d)) continue;
if (res(i,a)) f1=1;
if (res(i,d)) f2=1;
for (j=1;j<=N;j++){
f3=0;f4=0;
if (!res(j,b)&&!res(j,e)) continue;
if (res(j,b)) f3=1;
if (res(j,e)) f4=1;
for (k=1;k<=N;k++){
if (f1&&f3&&res(k,c))  r++;
else if (f2&&f4&&res(k,f))  r++;//避免重复
}
}
}
fout<<r<<endl;
return 0;
}

感觉自己之前写的逻辑不是很好，说一下我现在的逻辑吧！

我理解的枚举分两种：

第一种是不管三七二十一把所有的情况都列出来，然后在判断是否满足给定的条件，

第二种是找到可能的解的区间。

像这一道题目我们还可以将上下两个列出来，但是发现还要去重，所以我果断选择了第一种。

枚举每一位可能出现的情况，如果它不和任意一个距离 ” 小于 ” 2的话就不可能了，需要我们注意的是，我们最后还需要判断一下是否三位同时满足两个中的某一个，因为我们只要某一位满足两个中的某一个就可以了，但是我们最后要求的是三位同时满足。

/*
ID: 18795871
PROG: combo
LANG: C++
*/
#include<iostream>
#include<fstream>
#include<cstring>
using namespace std;

ifstream fin("combo.in");
ofstream fout("combo.out");

int n,cnt;
int x1[3],x2[3];

bool isOK1(int p,int w){
if (p==x1[w]) return true;
if (x1[w]==2){
if (p==1||p==n||p==3||p==4) return true;
return false;
}
if (x1[w]==1){
if (p==n-1||p==n||p==2||p==3) return true;
return false;
}
if (x1[w]==n){
if (p==n-2||p==n-1||p==1||p==2) return true;
return false;
}
if (x1[w]==n-1){
if (p==n-2||p==n-3||p==n||p==1) return true;
return false;
}
if (p<=x1[w]+2&&p>=x1[w]-2) return true;
return false;
}

bool isOK2(int p,int w){
if (p==x2[w]) return true;
if (x2[w]==2){
if (p==1||p==n||p==3||p==4) return true;
return false;
}
if (x2[w]==1){
if (p==n-1||p==n||p==2||p==3) return true;
return false;
}
if (x2[w]==n){
if (p==n-2||p==n-1||p==1||p==2) return true;
return false;
}
if (x2[w]==n-
4000
1){
if (p==n-2||p==n-3||p==n||p==1) return true;
return false;
}
if (p<=x2[w]+2&&p>=x2[w]-2) return true;
return false;
}

int main(){
while (fin>>n){
cnt=0;
int i,j,k;
fin>>x1[0]>>x1[1]>>x1[2]>>x2[0]>>x2[1]>>x2[2];
for (i=1;i<=n;i++){
if ((!isOK1(i,0))&&(!isOK2(i,0))) continue;
for (j=1;j<=n;j++){
if ((!isOK1(j,1))&&(!isOK2(j,1))) continue;
for (k=1;k<=n;k++){
if ((!isOK1(k,2))&&(!isOK2(k,2))) continue;
if ((isOK1(i,0)&&isOK1(j,1)&&isOK1(k,2))||(isOK2(i,0)&&isOK2(j,1)&&isOK2(k,2))){
cnt++;
}
}
}
}
fout<<cnt<<endl;
}
return 0;
}

但是我的逻辑函数写的太不优美了！于是乎，出现了最后的代码：

/*
ID: 18795871
PROG: combo
LANG: C++
*/
#include<iostream>
#include<fstream>
#include<cmath>
using namespace std;

ifstream fin("combo.in");
ofstream fout("combo.out");

int n,cnt;
int x1[3],x2[3];

bool isOK1(int p,int w){
if (abs(p-x1[w])<=2||abs(p-x1[w])>=n-2) return true;
return false;
}

bool isOK2(int p,int w){
if (abs(p-x2[w])<=2||abs(p-x2[w])>=n-2) return true;
return false;
}

int main(){
while (fin>>n){
cnt=0;
int i,j,k;
fin>>x1[0]>>x1[1]>>x1[2]>>x2[0]>>x2[1]>>x2[2];
for (i=1;i<=n;i++){
if ((!isOK1(i,0))&&(!isOK2(i,0))) continue;
for (j=1;j<=n;j++){
if ((!isOK1(j,1))&&(!isOK2(j,1))) continue;
for (k=1;k<=n;k++){
if ((!isOK1(k,2))&&(!isOK2(k,2))) continue;
if ((isOK1(i,0)&&isOK1(j,1)&&isOK1(k,2))||(isOK2(i,0)&&isOK2(j,1)&&isOK2(k,2))){
cnt++;
}
}
}
}
fout<<cnt<<endl;
}
return 0;
}