Deltix Round, Autumn 2021 (open for everyone, rated, Div. 1 + Div. 2) C、D题解

C. Complex Market Analysis

题意：
给定数列a，求出满足以下条件的数列的个数

思路：
我们可以从每个数的贡献来考虑。因为要求数列乘积为素数，那么只有素数才能对答案有所贡献，且与其相乘的数必定都为1且连续，只有这样数列乘积才是素数。因为从ai开始每次下标都增加e，那么下标index%e不相同的数一定互不影响，所以我们可以将index%e相同的数放在一起考虑，每次只考虑index%e相同的一组即可。我们定义l[i]表示：i节点左边连续1的个数，r[i]表示：i节点右边连续1的个数，这里的连续1都是对于同一组的数而言（i%e==j%e，即表示i、j在同一组）。

最后计算每个数对答案的贡献即可：

1. a[i]不是素数，对答案没有贡献
2. a[i]是素数，其贡献是：l[i]+r[i]+l[i]*r[i]

比如：1 2 1 1这个序列，l[2]=1,r[2]=2;
贡献=只取左边（1 2）+只取右边（2 1、2 1 1）+左右组合取（1 2 1、1 2 1 1）=5

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
const int inf=1e6+5;

int a[maxn];
ll l[maxn],r[maxn];

int is_prime[inf]; 
int prime[inf];
int cnt=0;

void getPrime(){//线性筛选素数 
	for(int i=2;i<inf;i++){
		if(!is_prime[i])	prime[++cnt]=i;
		for(int j=1;j<=cnt&&i*prime[j]<inf;j++){
			is_prime[i*prime[j]]=1;
			if(i%prime[j]==0)	break;
		}
	}
	is_prime[1]=1;
}

int main(){
	ios::sync_with_stdio(0);
	
	getPrime();
	int t;
	cin>>t;
	while(t--){
		int n,e;
		cin>>n>>e;
		for(int i=1;i<=n;i++)	cin>>a[i];
		ll ans=0;
		int num=0;//当前连续1数 
		for(int i=1;i<=e;i++){
			int j;
			num=0;
			for(j=i;j<=n;j+=e){
				if(a[j]==1)	num++;
				else{
					l[j]=num;
					num=0;
				}
			}
			num=0;
			for(j-=e;j>=i;j-=e){
				if(a[j]==1)	num++;
				else{
					r[j]=num;
					num=0;
				}
			}
		}
		for(int i=1;i<=n;i++)	if(!is_prime[a[i]])	ans+=l[i]+r[i]+l[i]*r[i];
		cout<<ans<<endl;
	}
	return 0;
}

D - Social Network

题意有点难懂：
给定n ，d。有d次关系检验，对于第i次，给定x，y，不管你怎么连线，只要确保x，y在同一块中且前i-1次关系也必须满足，每次输出最大块中元素数-1即可，注意每次i都是独立的（即在i次之前做的连线都可以改变）。

思路：
并查集+贪心。
对每次检验，当x,y分别属于两个不同的块时，将两个块合并并更新块的大小即可；当x,y已经在同一个块时，不用再将x、y合并，表明此时有一个灵活操作，可以合并任意两块，贪心的想一定合并最大的块。因为每次检验i都是独立的，所以，考虑在i次检验时，灵活操作的数量cnt，只要将cnt个最大的块相加-1即是第i次的答案。
时间复杂度O(n^2)

#include<bits/stdc++.h>
using namespace std;
const int maxn=1005;

int f[maxn];
int num[maxn];//num[i]:表示i所在块的元素的数量 
int n,d,ans=0;
int cnt=1;//取最大的cnt块相加，含义：可以灵活操作的个数+1 
vector<int> v;

int cmp(int x,int y){
	return x>y;
}

int getf(int x,int add){
	if(f[x]==x){
		num[x]+=add;
		ans=max(ans,num[x]);
		return x;
	}
	else{
		f[x]=getf(f[x],add);
		num[x]=num[f[x]]; 
		return f[x];
	}
}

void merge(int x,int y){
	v.clear();
	int ans=0;
	int xfa=getf(x,0);
	int yfa=getf(y,0);
	if(xfa!=yfa){
		f[xfa]=yfa;
		getf(f[x],num[xfa]);//合并两个块，并更新此块的num数量 
	}else	cnt++;//已经在一个块中，那么此次操作可以灵活操作 
	
	//取cnt个最大的块相加
	for(int i=1;i<=n;i++)	if(f[i]==i)	v.push_back(num[i]);//这样时间复杂度O(n^2)，因为n=1000，所以能过 
															//但是应该可以优化。 
	sort(v.begin(),v.end(),cmp);
	for(int i=0;i<cnt;i++)	ans+=v[i];
	cout<<ans-1<<endl;
}

int main(){
	ios::sync_with_stdio(0);
	
	cin>>n>>d;
	for(int i=1;i<=n;i++)	f[i]=i,num[i]=1;//初始化 
	for(int i=1;i<=d;i++){
		int x,y;
		cin>>x>>y;
		merge(x,y);
	}
	return 0;
}
/*
7 6
1 2
3 4
2 4
7 6
6 5
1 7
*/