Codeforces round 875 div2 A-D（D后补）

比赛链接

A.Twin Permutations（构造）

题意，给定一个排列a，要你构造一个排列b，使得ai+bi是不减的，取bi=n+1-ai即可。

B.Array merging（模拟）

题意：给定两个序列a,b，从a中取一段连续的颜色段，从b中取一段连续的颜色段，使拼接起来的颜色段长度最大。ai,bi<=2*n。所以可以考虑暴力枚举a，b中的最长颜色段，复杂度O()。注意一些细节即可。

#include<iostream>
#include<algorithm>
using namespace std;
const int N=2e5+10;
int T,n,a[N],b[N],cnt[N*2][2];
void check(){
	for(int i=1;i<=2*n;i++)cout<<cnt[i][0]<<' ';
	cout<<endl;;
	for(int i=1;i<=n+n;i++)cout<<cnt[i][1]<<' ';
	cout<<endl;
}
void solve(){
	cin>>n;
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1;i<=n;i++)cin>>b[i];
	//考虑每种颜色的最长子段
	for(int i=0;i<=2*n;i++)cnt[i][0]=cnt[i][1]=0;
	int now=a[1];int res=1;
	for(int i=2;i<=n;i++){
		if(a[i]==now){
			res++;
			continue;
		}
		else{
			cnt[a[i-1]][0]=max(cnt[a[i-1]][0],res);
			res=1;
			now=a[i];
		}
	}
	cnt[a[n]][0]=max(cnt[a[n]][0],res);
	now=b[1]; res=1;int ans=0;
	for(int i=2;i<=n;i++){
		if(b[i]==now){
			res++;
			continue;
		}
		else{
			cnt[b[i-1]][1]=max(cnt[b[i-1]][1],res);//死因：我把这里的1写成了0
			res=1;
			now=b[i];
		}
	}
	cnt[b[n]][1]=max(cnt[b[n]][1],res);
	//check();
	for(int i=0;i<=n*2;i++)ans=max(ans,cnt[i][0]+cnt[i][1]);
	cout<<ans<<endl;
}
int main(){
	cin>>T;
	while(T--)solve();
	return 0;
}

C.Copil Copac Draws Trees(思维，dfs)

题意：给定一个树，按照一定的步骤画这颗树。如果暴力枚举，最坏的时间复杂度是O（n^2），考虑如何优化

不难发现，画某条边的时间就是读入的顺序。

考虑维护当前时间戳，若边的时间戳＞当前时间戳则能继续画，否则需要进入下一次（下一层再画）

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=2e5+10;
int T,n,a[N],ans,head[N],len,vis[N];
struct edge{
	int v,w,next;
	edge(){
		
	}
	edge(int a,int b,int c){
		v=a;
		w=b;
		next=c;
	}
}e[N<<1];
void init(){
	len=0;
	memset(head,-1,sizeof head);
}
void add(int u,int v,int w){
	e[len]=edge(v,w,head[u]);
	head[u]=len++;
}
void dfs(int u,int f,int p,int t){
	for(int i=head[u];~i;i=e[i].next){
		int v=e[i].v;
		int w=e[i].w;
		if(v==f)continue;
		if(w<p){
			ans=max(ans,t+1);
			dfs(v,u,w,t+1);//w记录当前能到的最大时间 t记录层数 
		}else{
			ans=max(ans,t);
			dfs(v,u,w,t);
		}
	}
}
void solve(){
	cin>>n;ans=0;
	init();
	for(int i=1;i<n;i++){
		int u,v;
		cin>>u>>v;
		add(u,v,i);
		add(v,u,i);//w是时间戳 
	}
	for(int i=1;i<n;i++)vis[i]=0;vis[1]=1;
	dfs(1,0,0,1);
	cout<<ans<<endl;
}
int main(){
	cin>>T;
	while(T--)solve();
	return 0;
}

DThe BOSS Can Count Pairs（补题）（思维）

题意：给定序列a,b，需要找满足ai*aj=bi+bj的(i,j)对数。

关键：a，b的数据范围。ai,bi<=n。

我们先排个序

那么ai*aj=bi+bj<=2*n，所以ai只需要考虑0~，枚举ai，aj，bi，顺便算出bj，累加答案对数即可。
 

#include<iostream>
#include<algorithm>
#include<cstring>
#define int long long 
using namespace std;
const int N=2e5+10;
pair<int,int>c[N];
int T,n,a[N],b[N],cnt[N];
void solve(){
	cin>>n;
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1;i<=n;i++)cin>>b[i];
	for(int i=1;i<=n;i++){
		c[i]={a[i],b[i]};
	}
	sort(c+1,c+n+1);//排序
	int ans=0;
	for(int i=1;i*i<=(n<<1);i++){//枚举aj<=2sqrt(n)
		memset(cnt,0,sizeof cnt); 
		for(int j=1;j<=n;j++){
			//ai*aj=bi+bj
			int x=i*c[j].first-c[j].second;
			if(x>=1&&x<=n)ans+=cnt[x];
			if(c[j].first==i)cnt[c[j].second]++;//枚举到了当前这个位置 
		}
	}
	cout<<ans<<endl;
}
signed main(){
	cin>>T;
	while(T--)solve();
	return 0;
}