GT and set

Accepts: 35

Submissions: 194

Time Limit: 2000/1000 MS (Java/Others)

Memory Limit: 65536/65536 K (Java/Others)

问题描述

有NN个集合，每个集合中有A_iA?i??个数。
你要将这NN个集合划成LL个部分，使得每个部分的集合至少有一个共有的数。
如果至少有一个解输出YESYES，否则输出NONO

输入描述

第一行一个数T表示数据组数。（TT\leq≤2020）

对于每一组数据：
第一行两个数NN和LL。
接下来NN行每行描述一个集合：
第一个数A_iA?i??表示该集合的大小，之后xx个互不相同的整数表示该集合的元素。
集合里的数字都是正整数且不大于300300.

1\leq1≤NN\leq30≤30，1\leq1≤L\leq5L≤5，1\leq1≤A_iA?i??\leq10≤10，1 \leq L \leq N1≤L≤N

hack时建议输出最后一行的行末回车;每一行的结尾不要输出空格。

输出描述

对于每组数据输出一行YESYES或NONO

输入样例

2
2 1
1 1
1 2
3 2
3 1 2 3
3 4 5 6
3 2 5 6

输出样例

NO
YES

Hint

对于第二个样例，有三个集合{1 2 3}，{4 5 6}，{2 5 6} 你要划成两个部分。
有一种方案是把第二个和第三个集合划成一个部分，第一个在另一个部分。有一种方案是把第二个和第三个集合划成一个部分，第一个在另一个部分。 第二个和第三个集合的数字有一个交集{6}，所以合法。
还有一种划分方案就是把第一个和第三个集合划成一个部分，第二个在另一个部分。

///1085422276
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<queue>
#include<cmath>
#include<map>
#include<bitset>
#include<set>
#include<vector>
using namespace std ;
typedef unsigned long long ll;
#define mem(a) memset(a,0,sizeof(a))
#define meminf(a) memset(a,127,sizeof(a));
#define memfy(a)  memset(a,-1,sizeof(a));
#define TS printf("111111\n");
#define FOR(i,a,b) for( int i=a;i<=b;i++)
#define FORJ(i,a,b) for(int i=a;i>=b;i--)

#define inf 100000000
inline ll read()
{
    ll x=0,f=1;
    char ch=getchar();
    while(ch<‘0‘||ch>‘9‘)
    {
        if(ch==‘-‘)f=-1;
        ch=getchar();
    }
    while(ch>=‘0‘&&ch<=‘9‘)
    {
        x=x*10+ch-‘0‘;
        ch=getchar();
    }
    return x*f;
}
//***************************************
#define maxn 33
bool flag,vis[maxn];
int num[maxn],n,A,L;
bitset<301> S[maxn],tmp,temp,aa[maxn];

void dfs(int x)
{

      if(x==n+1)
      {
          flag=1;
          for(int i=1;i<=L;i++)
          {
              if(!num[i])flag=0;
          }
          return ;
      }
        if(flag)return ;
      for(int i=1;i<=L;i++){
        temp=aa[i];
          tmp=aa[i]&S[x];
          if(tmp.count()!=0)
          {
              aa[i]=tmp;
              num[i]++;
              dfs(x+1);
              num[i]--;
              aa[i]=temp;
          }
      }

}
int main()
{

    int T=read();
    while(T--)
    {
        scanf("%d%d",&n,&L);
        mem(vis);mem(num);
        FOR(i,1,30){
            S[i].reset();aa[i].set();
        }
        FOR(i,1,n){
            scanf("%d",&A);
            FOR(j,1,A){
                int x=read();
                S[i].set(x);
            }
        }
    //    for(int i=1;i<=6;i++)cout<<S[3][i];
        flag=0;
        dfs(1);
        if(flag){
            cout<<"YES"<<endl;
        }
        else{
            cout<<"NO"<<endl;
        }
    }
    return 0;
}

代码