poj 1659 Frogs' Neighborhood （DFS）

时间：2021-03-01 03:16:53

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

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

Description

未名湖附近共有N个大小湖泊L1, L2, ..., Ln(其中包括未名湖)，每个湖泊Li里住着一只青蛙Fi(1 ≤ i ≤ N)。如果湖泊Li和Lj之间有水路相连，则青蛙Fi和Fj互称为邻居。现在已知每只青蛙的邻居数目x1, x2, ..., xn，请你给出每两个湖泊之间的相连关系。

Input

第一行是测试数据的组数T(0 ≤ T ≤ 20)。每组数据包括两行，第一行是整数N(2 < N < 10)，第二行是N个整数，x1, x2,..., xn(0 ≤ xi ≤ 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

 
【题解】：直接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 ;

 }

相关文章