UVA - 167

UVA - 167
八皇后問題，用c[i]=k表示第i行放置第k列。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=25;
int mapp[maxn][maxn];
int ans;
int c[maxn];
void dfs(int cur,int sum)
{
    if(cur==8)
    {
        if(sum>ans)
            ans=sum;
        return ;
    }
    for(int i=0;i<8;i++)
    {
        int flag=0;
        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])
            {
                flag=1;
                break;
            }
        }
        if(!flag) dfs(cur+1,sum+mapp[cur][i]);
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(c,-1,sizeof(c));
        for(int i=0;i<8;i++)
            for(int j=0;j<8;j++)
            scanf("%d",&mapp[i][j]);
        ans=0;
        dfs(0,0);
        printf("%5d\n",ans);
    }
    return 0;
}

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=25;
int mapp[maxn][maxn];
int ans;
int c[maxn];
int vis[5][maxn];
void dfs(int cur,int sum)
{
    if(cur==8)
    {
        if(sum>ans)
            ans=sum;
        return ;
    }
    for(int i=0;i<8;i++)
    {
        if(!vis[0][i]&&!vis[1][cur+i]&&!vis[2][cur-i+8])
        {
            c[cur]=1;
            vis[0][i]=vis[1][cur+i]=vis[2][cur-i+8]=1;
            dfs(cur+1,sum+mapp[cur][i]);
            vis[0][i]=vis[1][cur+i]=vis[2][cur-i+8]=0;
        }
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(vis,0,sizeof(vis));
        memset(c,-1,sizeof(c));
        for(int i=0;i<8;i++)
            for(int j=0;j<8;j++)
            scanf("%d",&mapp[i][j]);
        ans=0;
        dfs(0,0);
        printf("%5d\n",ans);
    }
    return 0;
}