【暑假】[深入动态规划]UVa 1627 Team them up!

UVa 1627 Team them up!

题目：

Team them up!

Time Limit: 3000MS

Memory Limit: Unknown

64bit IO Format: %lld & %llu

Submit Status

Description

Your task is to divide a number of persons into two teams, in such a way, that:

everyone belongs to one of the teams;
every team has at least one member;
every person in the team knows every other person in his team;
teams are as close in their sizes as possible.

This task may have many solutions. You are to find and output any solution, or to report that the solution does not exist.

Input

The input begins with a single positive integer on a line by itself indicating the number of the cases following, each of them as described below. This line is followed by a blank line, and there is also a blank line between two consecutive inputs.

For simplicity, all persons are assigned a unique integer identifier from 1 to N.

The first line in the input file contains a single integer number N (2 ≤ N ≤ 100) - the total number of persons to divide into teams, followed by N lines - one line per person in ascending order of their identifiers. Each line contains the list of distinct numbers A_ij (1 ≤ A_ij ≤ N, A_ij ≠ i) separated by spaces. The list represents identifiers of persons that i^th person knows. The list is terminated by 0.

Output

For each test case, the output must follow the description below. The outputs of two consecutive cases will be separated by a blank line.

If the solution to the problem does not exist, then write a single message "No solution" (without quotes) to the output file. Otherwise write a solution on two lines. On the first line of the output file write the number of persons in the first team, followed by the identifiers of persons in the first team, placing one space before each identifier. On the second line describe the second team in the same way. You may write teams and identifiers of persons in a team in any order.

Sample Input

Sample Output

No solution

3 1 3 5

2 2 4

--------------------------------------------------------------------------------------------------------------------------------------------------------------------

思路：

给出关系图，不相识（互相）的两人必须分在不同组，要求分成两组且分组后有两组人数相差最少。

按照相反关系重新建图，如果两人不互相认识则连边，那么在一个联通块中，如何分组或是不能分组可知。如果不能构成二分图，那么问题无解因为不能满足必须分在不同组的要求。

设d[i][j+n]表示已经考虑到第i个联通块且两组相差i的情况是否存在。因为 j 属于[-n,n]所以需要+n调节j的范围。

有状态转移方程：

if(d[i][j+n])

d[i+1][j+n+diff[i]]=1;

d[i+1][j+n-diff[i]]=1;

其中diff[i]代表第i个联通块可分成的两组人数之差。

ans的得到需要按绝对值从小到大依此枚举，根据d[][]判断是否存在即可。

代码：

 #include<cstdio>

 #include<cstring>

 #include<vector>

 #define FOR(a,b,c) for(int a=(b);a<(c);a++)

 using namespace std;

 const int maxn =  + ;

 int colors_num,n,m;

 int d[maxn][*maxn],diff[maxn];

 int G[maxn][maxn];

 vector<int> team[maxn][];

 int colors[maxn];

 //如果不是二部图return false

 bool dfs(int u,int c) {

     colors[u]=c;  //c==1 || 2

     team[colors_num][c-].push_back(u);

     FOR(v,,n)

     if(u!=v && !(G[u][v]&&G[v][u])){ //不互相认识

         if(colors[v]> && colors[u]==colors[v]) return false;

         //u v不能在一组却出现在了一组

         if(!colors[v] && !dfs(v,-c)) return false;

     }

     return true;

 }

 bool build_graph() {

     colors_num=;

     memset(colors,,sizeof(colors));

     FOR(i,,n) if(!colors[i]){

         team[colors_num][].clear();

         team[colors_num][].clear();

         if(!dfs(i,)) return false;

         diff[colors_num]=team[colors_num][].size()-team[colors_num][].size();

         colors_num++;

     }

     return true;

 }

 void print(int ans) {

   vector<int> team1, team2;

   for(int i = colors_num-; i >= ; i--) {  //对 每个联通块

     int t;

     if(d[i][ans-diff[i]+n]) { t = ; ans -= diff[i]; }  //判断+- //组号为t

     else { t = ; ans += diff[i]; }

     for(int j = ; j < team[i][t].size(); j++)  //加入team1

       team1.push_back(team[i][t][j]);

     for(int j = ; j < team[i][^t].size(); j++) //加入team2

       team2.push_back(team[i][^t][j]);

   }

   printf("%d", team1.size());

   for(int i = ; i < team1.size(); i++) printf(" %d", team1[i]+);

   printf("\n");

   printf("%d", team2.size());

   for(int i = ; i < team2.size(); i++) printf(" %d", team2[i]+);

   printf("\n");

 }

 void dp() {

     //d[i][j+n] 代表考虑到第i个联通块时两组相差j的情况是否存在

     memset(d,,sizeof(d));

     d[][+n]=;

     //+n 调节范围

     FOR(i,,colors_num)

       FOR(j,-n,n+) if(d[i][j+n]) {

           //刷表 存在

           d[i+][j+n+diff[i]]=;

           d[i+][j+n-diff[i]]=;

       }

     FOR(ans,,n+) {

         if(d[colors_num][n+ans]) {print(ans); return; }

         if(d[colors_num][n-ans]) {print(-ans); return; }

     }

 }

 int main() {

   int T;  scanf("%d",&T);

   while(T--) {

       scanf("%d",&n);

       FOR(u,,n) {  //读入原图

           int v;

           while(scanf("%d",&v) && v) G[u][v-]=; //v-1调节序号

       }

       if(n== || !build_graph()) printf("No solution\n"); //n==1 -> no solution

       else  dp();

       if(T) printf("\n");

   }

   return ;

 }