Description
有n只怪兽,将其分成两堆T1和T2,给出T1的m只怪兽编号,给出一个关系矩阵G,Gij=1表示i怪兽比j怪兽强壮,在一个怪兽集合中,如果任意一只怪兽可以打败所有比其编号大的怪兽,那么这个集合合法,现在判断T1和T2是否合法,如果都合法问最多从T2中拿出多少只怪兽加到T1中使得T1仍然合法
Input
多组用例,每组用例首先输入两个整数n和m分别表示怪兽个数和T1中怪兽个数,之后一个n*n的01矩阵G表示怪兽之间的关系矩阵,最后m个整数表示T1中的怪兽编号,以0 0结束输入(2<=n<=1000,1<=m< n)
Output
对于每组用例,如果T1和T2都合法则输出YES并输出最多从T2中拿出多少只怪兽加到T1中使得T1仍然合法,否则输出NO
Sample Input
3 2
0 1 1
0 0 1
0 0 0
3 1
4 3
0 1 0 1
0 0 1 1
1 0 0 1
0 0 0 0
1 2 3
4 2
0 1 0 1
0 0 1 1
1 0 0 1
0 0 0 0
1 2
0 0
Sample Output
YES 1
NO
YES 1
Solution
一个合法集合其实就是一个竞赛图,例如判断T1是否合法,只需要记录T1中每个点的出度,如果这些出度恰好是一个0~m-1的重排那么说明T1是一个竞赛图,T2同理,判断完后显然可以得到T1和T2的拓扑序,至于第二个问题,对于T2中每只怪兽,如果其能加到T1中那么一定可以找到其在T1的拓扑序中的位置,即前面的怪兽都能打败这个怪兽,后面的怪兽都能被这只怪兽打败(如果找不到说明这只怪兽不能被加到T1中),以这个位置作为这n-m只怪兽的权值,求一遍最长上升子序列,这个LIS的长度即为答案,时间复杂度O(n^2)
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define INF 0x7fffffff
#define maxn 1111
int n,m,g[maxn][maxn],vis[maxn],c[maxn];
struct node
{
int id,v;
bool operator <(const node &b)const
{
return v>b.v;
}
}a[maxn],b[maxn];
int dp[maxn];
int LIS(int a[],int n)//求序列a的(非严格)最长上升子序列
{
if(n==0)return 0;
for(int i=1;i<n;i++)dp[i]=INF;
dp[0]=a[0];
int len=1;
for(int i=1;i<n;i++)
{
if(a[i]>=dp[len-1])
dp[len++]=a[i];
else
dp[upper_bound(dp,dp+n,a[i])-dp]=a[i];
}
return len;
}
char s[2*maxn];
int main()
{
while(~scanf("%d%d\n",&n,&m),n||m)
{
for(int i=1;i<=n;i++)
{
gets(s);
for(int j=1;j<=n;j++)g[i][j]=s[2*j-2]-'0';
}
int flag=1;
memset(vis,0,sizeof(vis));
for(int i=1;i<=m;i++)scanf("%d",&a[i].id),vis[a[i].id]=1;
for(int i=1,j=0;i<=n;i++)if(!vis[i])b[++j].id=i;
memset(vis,0,sizeof(vis));
for(int i=1;i<=m;i++)
{
int cnt=0;
for(int j=1;j<=m;j++)
if(i!=j&&g[a[i].id][a[j].id])cnt++;
a[i].v=cnt;
vis[cnt]=1;
}
for(int i=0;i<m;i++)
if(!vis[i])
{
flag=0;
break;
}
memset(vis,0,sizeof(vis));
for(int i=1;i<=n-m;i++)
{
int cnt=0;
for(int j=1;j<=n-m;j++)
if(i!=j&&g[b[i].id][b[j].id])cnt++;
b[i].v=cnt;
vis[cnt]=1;
}
for(int i=0;i<n-m;i++)
if(!vis[i])
{
flag=0;
break;
}
if(!flag)printf("NO\n");
else
{
printf("YES ");
sort(a+1,a+m+1);
sort(b+1,b+n-m+1);
int res=0;
for(int i=1;i<=n-m;i++)
{
int l=1,r=m+1;
for(int j=1;j<=m;j++)
if(g[b[i].id][a[j].id])r=min(r,j);
else l=max(l,j+1);
if(l==r)c[res++]=l;
}
int ans=LIS(c,res);
printf("%d\n",ans);
}
}
return 0;
}