回溯例题（计算机算法设计与分析）

P153：子集和问题

#include <bits/stdc++.h>

using namespace std;

int n,s;
bool vis[100];
int a[100];
bool calc(int cur) {
    if(cur==s) {
        for(int i=0;i<n;i++) {
            if(vis[i])
                printf("%d ",a[i]);
        }
        puts("");
        return true;
    }
    else {
        for(int i=0;i<n;i++) {
            if(!vis[i]) {
                vis[i] = true;
                if(calc(cur+a[i]))
                    return true;
                vis[i] = false;
            }
        }
    }
    return false;
}

int main()
{
    freopen("in.txt","r",stdin);
    scanf("%d%d",&n,&s);
    for(int i=0;i<n;i++) {
        scanf("%d",&a[i]);
    }
    if(!calc(0))
        puts("No Solution!");
    return 0;
}

 

P152：最小重量机器设计问题

#include <bits/stdc++.h>

using namespace std;

int n,m,c;
const int maxn = 100;

struct Node {
    int w;
    int c;
    int id;
    bool operator < (const Node & rhs) const {
        if(w!=rhs.w)
            return w < rhs.w;
        else return c < rhs.c;
    }
};

vector<Node> nn[maxn];

int ccc[maxn][maxn];
int www[maxn][maxn];
bool vis[maxn][maxn];

bool calc(int cur,int cc) {
    if(cur>=n&&cc<=c) {
        printf("%d\n",cc);
        for(int i=0;i<n;i++) {
            for(int j=0;j<m;j++) {
                if(vis[i][j]) {
                    printf("%d ",nn[i][j].id);
                    break;
                }
            }
        }
        puts("");
        return true;
    }

    else {
        for(int i=0;i<m;i++) {
            vis[cur][i] = true;
            int cx = cc+nn[cur][i].c;
            if(calc(cur+1,cx))
                return true;
            vis[cur][i] = false;
        }
    }
    return false;
}


int main()
{
    freopen("in.txt","r",stdin);
    scanf("%d%d%d",&n,&m,&c);
    for(int i=0;i<n;i++) {
        for(int j=0;j<m;j++) {
            scanf("%d",&ccc[i][j]);
        }
    }

    for(int i=0;i<n;i++) {
        for(int j=0;j<m;j++) {
            scanf("%d",&www[i][j]);
        }
    }

    for(int i=0;i<n;i++) {
        for(int j=0;j<m;j++) {
            nn[i].push_back((Node){www[i][j],ccc[i][j],j+1});
        }
        sort(nn[i].begin(),nn[i].end());
    }

    calc(0,0);

    return 0;
}

 

P152：运动员最佳配对问题

#include <bits/stdc++.h>

using namespace std;

const int maxn = 100;

int n;
int p[maxn][maxn];
int q[maxn][maxn];
bool vis[maxn];

int ans;
void calc(int cur,int cc) {
    if(cur>n) {
        ans = max(ans,cc);
    }
    for(int i=0;i<n;i++) {
        if(!vis[i]) {
            vis[i] = true;
            calc(cur+1,cc+p[cur][i]*q[i][cur]);
            vis[i] = false;
        }
    }
}

int main()
{
    freopen("in.txt","r",stdin);
    scanf("%d",&n);
    for(int i=0;i<n;i++) {
        for(int j=0;j<n;j++) {
            scanf("%d",&p[i][j]);
        }
    }
    
    for(int i=0;i<n;i++) {
        for(int j=0;j<n;j++) {
            scanf("%d",&q[i][j]);
        }
    }
    
    calc(0,0);
    printf("%d\n",ans);
    
    return 0;
}

 

P196：n皇后问题

#include <bits/stdc++.h>

using namespace std;

const int maxn = 25;
int C[maxn];
int n;
void calc(int cur) {
    if(cur>=n) {
        for(int i=0;i<n;i++)
            printf("%d ",C[i]+1);
        puts("");
        return;
    }
    else {
        for(int i=0;i<n;i++) {
            int ok = 1;
            C[cur] = i;
            for(int j=0;j<cur;j++) {
                if(C[cur]==C[j]||cur-C[cur]==j-C[j]||cur+C[cur]==j+C[j]) {
                    ok = 0;
                    break;
                }
            }
            if(ok) {
                calc(cur+1);
                return;
            }
        }
    }
    return;
}

int main()
{
    scanf("%d",&n);
    calc(0);
    return 0;
}