AtCoder Regular Contest 097 题解

D - Equals：

并查集维护$a[i],i$是否联通。
显然，只要联通就一定可以换到。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
int n,m,a[100010],fa[100010];
int findfa(int x) {return fa[x]==x?x:fa[x]=findfa(fa[x]);}
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),fa[i]=i;
	for(int i=1;i<=m;i++)
	{
		int x,y;scanf("%d %d",&x,&y);
		int tx=findfa(x),ty=findfa(y);
		fa[tx]=ty;
	}
	int ans;
	for(int i=1;i<=n;i++) if(findfa(a[i])==findfa(i)) ans++;
	printf("%d",ans);
}

E - Sorted and Sorted：

假如只有一个排列就是经典问题，答案就是逆序对的个数，因为每次只能交换一个逆序对。
这题将两个序列混在一起，那么重新定义逆序对的概念：假如原序列$a_i,a_j$满足$a_i$在$a_j$前面且最终序列中$a_i$在$a_j$后面，那么就是一对逆序对。这样答案还是序列的逆序对数。
于是可以dp，$f[i][j]$表示已经排好$1$ ~ $i$,$1$~$j$的最小代价（先不管其他数），预处理后可以$O(1)$转移。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
using namespace std;
LL C[3010][3010],S[3010][3010];
LL n,p;
void pre()
{
	for(LL i=0;i<=3001;i++)
	{
		C[i][0]=1;
		for(LL j=1;j<=i;j++) C[i][j]=(C[i-1][j]+C[i-1][j-1])%p;
	}
	S[0][0]=1;
	for(LL i=1;i<=3001;i++)
	{
		S[i][1]=1;
		for(LL j=2;j<=i;j++) S[i][j]=(S[i-1][j]*j%p+S[i-1][j-1])%p;
	}
}
LL pow(LL a,LL b,LL mod)
{
	LL ans=1;
	while(b)
	{
		if(b&1) ans=ans*a%mod;
		a=a*a%mod;b>>=1;
	}
	return ans;
}
int main()
{
	LL ans=0;scanf("%lld %lld",&n,&p);
	pre();
	for(LL i=0;i<=n;i++)
	{
		LL s=C[n][i]*pow(2,pow(2,n-i,p-1),p)%p;
		LL tot=0;
		for(LL j=0;j<=i;j++) tot=(tot+pow(2,(n-i)*j%(p-1),p)*S[i+1][j+1]%p)%p;
		ans=(ans+(i&1?-1:1)*s%p*tot%p)%p;
	}
	printf("%lld",(ans+p)%p);
}

F - Monochrome Cat：

这道题应该暴力的dp挺直观的，但是正解很简洁巧妙
首先将所有黑色叶子节点（子树）删掉，显然一辈子也不会到哪里去。
首先假设这只猫要走一个回路，就是起点终点相同（从任何一个点出发都是一样的），那么每条边要走两次，每个点的访问次数是确定的，所以对于每个点，是否要进行2操作也是知道的。
当起点终点不一样的时候，显然只有$S->T$的点会受影响。如果一个点进行了2操作，因为少到一次，就不用做了，总操作次数-2；如果一个点没进行2操作，因为少到一次，就需要补一次，总操作次数不变。所以就成了树上带权直径。

#include<queue>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
struct node{
	int y,next;
}e[200010];int len=0,last[100010];
int n,c[100010],d[100010],f[100010];
bool vis[100010];
int tot=0,ans=0;
void ins(int x,int y)
{
	e[++len].y=y;
	e[len].next=last[x];last[x]=len;
}
char s[100010];
int N;
queue<int> q;
void pre()
{
	for(int i=1;i<=n;i++) if(s[i]=='B'&&d[i]==1) q.push(i);
	while(!q.empty())
	{
		int x=q.front();q.pop();
		vis[x]=true;N--;
		for(int i=last[x];i;i=e[i].next)
		{
			int y=e[i].y;
			d[y]--;if(d[y]==1&&s[y]=='B') q.push(y);
		}
	}
	if(N==0) {printf("0");exit(0);}
}
void dfs(int x,int fa)
{
	if(vis[x]) return;
	f[x]=c[x];ans=max(ans,c[x]);
	for(int i=last[x];i;i=e[i].next)
	{
		int y=e[i].y;
		if(y==fa) continue;
		dfs(y,x);ans=max(ans,f[x]+f[y]);
		f[x]=max(f[x],f[y]+c[x]);
	}
}
int main()
{
	scanf("%d",&n);N=n;
	for(int i=1;i<n;i++)
	{
		int x,y;scanf("%d %d",&x,&y);
		ins(x,y);ins(y,x);
		d[x]++;d[y]++;
	}
	memset(vis,false,sizeof(vis));
	scanf("%s",s+1);
	pre();
	for(int i=1;i<=n;i++) if(!vis[i]) tot+=d[i];
	for(int i=1;i<=n;i++)
	{
		if(vis[i]) continue;
		int k=s[i]=='B';
		if((d[i]+k)%2==1) c[i]=0;
		else tot++,c[i]=2;
	}
	if(N==1) {printf("%d",tot);return 0;}
	for(int i=1;i<=n;i++)
		if(!vis[i])
		{
			dfs(i,0);
			break;
		}
	printf("%d",tot-ans);
}