题面:
题目描述
给你一个 n×n 的矩阵 b。
你需要构造 n 个集合,集合里出现的数只能是 1∼n。
若 bi,j=1,则说明集合 i 是集合 j 的子集。否则说明集合 i 不是集合 j 的子集。
数据保证有解,你需要构造出这 n 个集合。
比如,
4 0001 1001 0001 0000
这道题起初想复杂了,我个人潜意识认为子集的子集 不是 子集,然后写了一个复杂的代码。
第一版代码,把每个集合置为下标,为了各异这一性质。用了深搜把当前集合子集的子集所有元素都加到res中,实则没必要,当前集合子集的子集这些都会在矩阵中体现。
#include <bits/stdc++.h>
#define endl '\n'
using namespace std;
const int N = 2e5 + 20;
typedef pair<int, int> PII;
typedef long long LL;
int g[110][110];
int n;
int dist[110];
void dfs(int u, vector<vector<int> > &res, vector<set<int> > &st)
{
for (int i = 1; i <= n; i ++)
{
if(g[u][i])
{
for (int j = 0; j < (int)res[u].size(); j ++)
{
int uop = res[u][j];
if(st[i].count(uop)) continue;
res[i].push_back(res[u][j]);
st[i].insert(uop);
}
dfs(i, res, st);
}
}
return ;
}
void solve()
{
cin >> n;
vector<vector<int> > res(n + 1);
vector<set<int> > st(n + 1);
for (int i = 1; i <= n; i ++)
{
st[i].insert(i);
res[i].push_back(i);
}
memset(dist, 0, sizeof dist);
for (int i = 1; i <= n; i ++)
{
cin.get();
char s;
for (int j = 1; j <= n; j ++)
{
cin >> s;
g[i][j] = s - '0';
if(g[i][j]) dist[j] ++;
}
}
for (int i = 1; i <= n; i ++)
{
if(!dist[i])
dfs(i, res, st);
}
for (int i = 1; i <= n; i ++)
{
/*
sort(res[i].begin(),res[i].end());//先排序1123345
res[i].erase(unique(res[i].begin(),res[i].end()),res[i].end());
*/
int m = res[i].size();
cout << m << " ";
for (int j = 0; j < m; j ++)
cout << res[i][j] << " ";
cout << endl;
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
int test = 1; cin >> test;
while(test --)
solve();
return 0;
}
虽然解法是对的,但是超时,因为有冗余遍历。
第二版,想的简单一些就好,不用考虑子集的子集这些问题,在矩阵中这些因素都会涵盖。
遍历矩阵的时候直接插入就好,不用向前向后深究。
#include<bits/stdc++.h>
using namespace std;
int t,n;
vector<int>q[105];
string s;
void inline slove()
{
cin >> n;
for(int i = 1; i <= n; i ++)
q[i].clear();
for(int i = 1; i <= n; i ++)
q[i].push_back(i);
for(int i = 1; i <= n; i ++)
{
cin>>s;
for(int j = 0; j < n; j ++)
{
if(s[j] == '1')
q[j + 1].push_back(i);
}
}
for(int i = 1; i <= n; i ++)
{
cout << q[i].size() << " ";
for(int j = 0; j < q[i].size(); j ++)
cout << q[i][j] << " ";
cout << endl;
}
}
int main(){
cin >> t;
while(t --)
slove();
}