I-Investigating Legions 2020牛客多校第4场

https://ac.nowcoder.com/acm/contest/5669/I

过了一车人，然而我哈不清be

乱搞思想是，随便从一个点i开始把所有是f[i][j]=1的点放到这个集合里面，然后枚举所有点j到这个集合中所有点相同的个数，如果cntj>=size/2那么j这个点就是跟i这个点一个集合的，很有道理的样子。

#include
#define pb push_back
using namespace std;
typedef long long ll;

const int maxl=510;

int n,S,m,cas,k,cnt,tot,ans;
int f[maxl][maxl],col[maxl],ini[maxl];
char s[maxl*maxl];
bool in[maxl]; 
vector tmp;

inline void prework()
{
	scanf("%d%d",&n,&S);
	scanf("%s",s+1);
	cnt=0;
	for(int i=1;i<=n;i++)
		for(int j=i+1;j<=n;j++)
			f[i][j]=f[j][i]=(s[++cnt]=='1');
} 

inline void mainwork()
{
	tot=0;
	for(int i=1;i<=n;i++)
		col[i]=0,f[i][i]=1;
	for(int i=1;i<=n;i++)
	if(!col[i])
	{
		tmp.clear();
		for(int j=1;j<=n;j++)
		if(!col[j] && f[i][j])
			tmp.pb(j);
		++tot;
		int len=tmp.size();
		for(int j=1;j<=n;j++)
		if(!col[j])
		{
			cnt=0;
			for(int x:tmp)
			if(f[x][j])
				cnt++;
			if(cnt>=len/2)
				col[j]=tot;
		}
	}
}

inline void print()
{
	for(int i=1;i<=n;i++)
		printf("%d ",col[i]-1);
	puts("");
}

int main()
{
	int t=1;
	scanf("%d",&t);
	for(cas=1;cas<=t;cas++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}