poj 1659 Frogs' Neighborhood （DFS）

http://poj.org/problem?id=1659

Frogs' Neighborhood

Time Limit: 5000MS		Memory Limit: 10000K
Total Submissions: 6050		Accepted: 2623		Special Judge

Description

未名湖附近共有N个大小湖泊L₁, L₂, ..., L_n(其中包括未名湖)，每个湖泊L_i里住着一只青蛙F_i(1 ≤ i ≤ N)。如果湖泊L_i和L_j之间有水路相连，则青蛙F_i和F_j互称为邻居。现在已知每只青蛙的邻居数目x₁, x₂, ..., x_n，请你给出每两个湖泊之间的相连关系。

Input

第一行是测试数据的组数T(0 ≤ T ≤ 20)。每组数据包括两行，第一行是整数N(2 < N < 10)，第二行是N个整数，x₁, x₂,..., x_n(0 ≤ x_i ≤ N)。

Output

对输入的每组测试数据，如果不存在可能的相连关系，输出"NO"。否则输出"YES"，并用N×N的矩阵表示湖泊间的相邻关系，即如果湖泊i与湖泊j之间有水路相连，则第i行的第j个数字为1，否则为0。每两个数字之间输出一个空格。如果存在多种可能，只需给出一种符合条件的情形。相邻两组测试数据之间输出一个空行。

Sample Input

3
7
4 3 1 5 4 2 1
6
4 3 1 4 2 0
6
2 3 1 1 2 1

Sample Output

YES
0 1 0 1 1 0 1
1 0 0 1 1 0 0
0 0 0 1 0 0 0
1 1 1 0 1 1 0
1 1 0 1 0 1 0
0 0 0 1 1 0 0
1 0 0 0 0 0 0 

NO

YES
0 1 0 0 1 0
1 0 0 1 1 0
0 0 0 0 0 1
0 1 0 0 0 0
1 1 0 0 0 0
0 0 1 0 0 0

Source

POJ Monthly--2004.05.15 Alcyone@pku

 

【题解】：直接DFS搜索  16ms 过

【code】：

 /**
 Satus:Accepted      Memory:676K
 Time:16MS     Language:G++
 Code Lenght:1774B  Author:cj
 */
 #include<iostream>
 #include<stdio.h>
 #include<algorithm>
 #include<vector>
 #include<queue>
 #include<string.h>

 #define N  11
 using namespace std;

 //map保存最终结果  vi表示已知i的邻居数  cnti表示i目前的邻居数目
 int map[N][N],v[N],cnt[N],n,flag;  //flag 标记 YES 或 NO

 void dfs(int id)  //对第id只青蛙进行搜索
 {
     int i,j;
     if(flag) return;  //找到了就直接退出
     if(id>n) //n只青蛙全部搜索完，表示已经出现结果了
     {
         flag =; //标记结果出现
         puts("YES");
         for(i=;i<=n;i++)
         {
             for(j=;j<=n;j++)
             {
                 if(j==)    printf("%d",map[i][j]|map[j][i]);  //用或运算将图的另一半对称输出
                 else    printf(" %d",map[i][j]|map[j][i]);  //用或运算将图的另一半对称输出
             }
             putchar(); //换行
         }
         return ;
     }
     if(cnt[id]==v[id])  //第id只青蛙已经搜索到满足条件的状态
     {
         dfs(id+);  //开始下一只青蛙的搜索
         return;
     }
     for(i=;i<=n;i++)  //只对图的右上半部分进行操作，另一半对称输出就行了
     {
         if(i<id)  //对y轴方向操作
         {
             if(!map[i][id]&&cnt[id]<v[id]&&cnt[i]<v[i])  //满足搜索条件记得加cnt[i]<v[i]
             {
                 map[i][id] = ;
                 cnt[id]++;
                 cnt[i]++;  //id与i两只青蛙邻居关系是相互的，所以要同时++
                 dfs(id);
                 cnt[i]--;
                 cnt[id]--;
                 map[i][id] = ;
             }
         }
         else if(i>id)  //对x轴方向操作
         {
             if(!map[id][i]&&cnt[id]<v[id]&&cnt[i]<v[i])
             {
                 map[id][i] = ;
                 cnt[id]++;
                 cnt[i]++;
                 dfs(id);
                 cnt[i]--;
                 cnt[id]--;
                 map[id][i] = ;
             }
         }
     }
 }

 int main()
 {
     int t,cas=;
     scanf("%d",&t);
     while(t--)
     {
         scanf("%d",&n);
         int i;
         for(i=;i<=n;i++)
         {
             scanf("%d",v+i);
         }
         memset(map,,sizeof(map));
         memset(cnt,,sizeof(cnt));
         flag = ;
         if(cas++)   putchar();
         dfs();  //从第一个青蛙开始搜索
         if(!flag)   puts("NO");
     }
     return ;
 }