题目大意:
有n 个人可能参加聚会,每个人参加聚会时会发短信给所有他认识的人
(包括他自己)。现在给定关系网(每个人认识哪些人,保证认识自己),
以及一个数列ai,请你给出一个去聚会的人的方案,使得最后所有人收
到的短信数bi 满足 bi ̸= ai,若无可行方案输出无解
实际上这个题没有无解的情况···
solution:
若目前所有ai 均> 0,则所有人不去就满足条件
若有ai = 0,考虑到题目满足所有人一定认识他自己,因此让i 这
个人参加即可使得bi ̸= ai
让第i 个人参加后,根据关系网更新每个人现在不应收到的短信数,
返回上面的步骤继续判断
不断重复这个过程,一定能找到一组解
O(n^2)
但这个实现起来还是需要一些技巧的
比如说如何不断找出ai=bi或者bi=0的呢,我们可以用队列!!
看代码特别像spfa···
可以把ai+1的过程改为bi-1,这样只需要每次bi=0时加入队列
还有为了方便,我们把人之间的关系存成vector
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#define maxn 105
using namespace std;
int n,b[maxn],cnt,ans[maxn];
char s[maxn];
bool flg;
vector<int> a[maxn];
queue<int> q;
inline int rd(){
int x=0,f=1; char c=' ';
while(c<'0' || c>'9') {if(c=='-') f=-1;c=getchar();}
while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
return x*f;
}
int main(){
n=rd();
for(int i=1;i<=n;i++){
scanf("%s",s+1);
for(int j=1;j<=n;j++)
if(s[j]=='1') a[i].push_back(j);
}
for(int i=1;i<=n;i++) {
b[i]=rd();
if(!b[i]) flg=true;
}
if(!flg){printf("0\n");return 0;}
for(int i=1;i<=n;i++) if(b[i]==0) q.push(i);
while(!q.empty()){
int x=q.front(); q.pop();
ans[++cnt]=x;
for(int i=0;i<a[x].size();i++){
int y=a[x][i];
b[y]--;
if(b[y]==0) q.push(y);
}
}
sort(ans+1,ans+cnt+1);
printf("%d\n",cnt);
for(int i=1;i<=cnt;i++) printf("%d ",ans[i]);
return 0;
}