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-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;
}