牛客挑战赛30 ABC题解

A. 小G数数

思路： 枚举下标 b c，然后用主席树快速求出[ 1 b-1 ]小于 A[c] 的元素个数t1，[ c+1 n ]大于A[ b ]的元素个数，答案加上t1*t2即可。

// 貌似发现了更高级的n^2做法，我就不做题解了

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=505;
ll ans;
int n,a[maxn],cnt,sum[maxn*10],vis[maxn][maxn];
int rt[maxn*10],ls[maxn*10],rs[maxn*10],mx;
void up(int pre,int& o,int l,int r,int k)
{
	o=++cnt;
	sum[o]=sum[pre]+1;
	ls[o]=ls[pre],rs[o]=rs[pre];
	if(l==r)return;
	int m=(l+r)/2;
	if(k<=m)up(ls[pre],ls[o],l,m,k);
	else up(rs[pre],rs[o],m+1,r,k);
}
int qu(int o,int l,int r,int ql,int qr)
{
	if(ql>qr)return 0;
	if(l>=ql&&r<=qr)return sum[o];
	int m=(l+r)/2,res=0;
	if(ql<=m)res+=qu(ls[o],l,m,ql,qr);
	if(qr>m)res+=qu(rs[o],m+1,r,ql,qr);
	return res;
}
void dfs(int l,int r)
{
	if(l==r||vis[l][r])return;
	vis[l][r]=1;
	dfs(l+1,r),dfs(l,r-1);
	if(a[l]<=a[r])return;
	int t1=qu(rt[l-1],1,mx,1,a[r]-1);
	int t2=qu(rt[n],1,mx,a[l]+1,n)-qu(rt[r],1,mx,a[l]+1,n);
	ans+=t1*t2;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]),mx=max(mx,a[i]);
	for(int i=1;i<=n;i++)
	up(rt[i-1],rt[i],1,mx,a[i]);
	dfs(1,n);
	printf("%lld\n",ans);
}

B. 小A的排列

思路：我的思路不怎么好，被卡常了，把很多取模优化和 ll 改成 int 勉强通过了。

枚举起点 i，我先把 [ i ,n ]的数用基数排序，然后按顺序用双向链表连接起来，用 o 代表其中间的那个数，然后枚举右边界 r 从n-->i，每次算好贡献后，把a[r]从链表中删除，然后根据a[r]和o的大小关系适当移动o使得o成为新链表的中间点即可。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=10005,mod=1e9+7;
int a[maxn],b[maxn],n;
int h[maxn],t[maxn];
int inv,k,ans;
int ksm(ll x,int y)
{
	ll res=1;
	while(y)
	{
		if(y&1)res=res*x%mod;
		x=x*x%mod;
		y/=2; 
	}
	return res;
}
void add(int& x,int y)
{
	if(1ll*x+y>=mod)x=x-mod+y;
	else x=x+y;
}
void gao(int cur)
{
	int m=n-cur+1,pre=0,last,o=0,cnt=0;
	for(int i=cur;i<=n;i++)b[a[i]]=cur;
	for(int i=1;i<=n;i++)
	if(b[i]==cur)
	{
		cnt++;
		if(cnt==(m+1)/2)o=i;
		t[pre]=i,h[i]=pre,pre=i;	
	}
	t[pre]=0;
	int res=ksm(k,(cur-1)*n+n);
	for(int i=n;i>=cur;i--)
	{
		if(m%2)add(ans,1ll*res*o*2%mod);
		else add(ans,1ll*res*(o+t[o])%mod);		
		pre=h[a[i]];
		last=t[a[i]];
		t[pre]=last;
		h[last]=pre;
		if(a[i]==o)
		{
			if(m%2)o=h[o];
			else o=t[o];
		}
		else if(a[i]<o)
		{
			if(m%2==0)o=t[o];
		}
		else
		{
			if(m%2)o=h[o];
		}
		m--;
		res=1ll*res*inv%mod;
	}
}
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	inv=ksm(k,mod-2);
	for(int i=1;i<=n;i++)
	gao(i);
	printf("%d\n",ans);
}

C. 小G砍树

思路：不难发现对于一棵树，如果最后删除的点是 o ，那么我们就假设该树以o为根节点，设sz[u]为u子树节点个数，显然删除方法一共有n!/sz[1]/sz[2]/.../sz[n]，然后我们枚举最后删除的点即可，就变成了换根dp，假设当前u为根，我要求最后删除点是v（u，v有边连接）的方式，显然d[v]=d[u]*sz[v]*inv[ n-sz[v] ]。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e5+10,mod=998244353;
ll ksm(ll x,int y)
{
	ll res=1;
	while(y)
	{
		if(y&1)res=res*x%mod;
		x=x*x%mod;
		y/=2;
	}
	return res;
}
ll d[maxn],inv[maxn],ans,sum;
int sz[maxn],n;
vector<int>G[maxn];
void mul(ll& x,ll y)
{
	x=x*y%mod;
}
void add(ll& x,ll y)
{
	x=(x+y)%mod;
}
void dfs(int u,int fa)
{
	sz[u]=1;
	for(int i=0;i<G[u].size();i++)
	{
		int v=G[u][i];
		if(v==fa)continue;
		dfs(v,u);
		sz[u]+=sz[v];
		mul(sum,inv[sz[v]]);
	}
}
void dfs2(int u,int fa,ll res)
{
	add(ans,res);
	for(int i=0;i<G[u].size();i++)
	{
		int v=G[u][i];
		if(v==fa)continue;
		dfs2(v,u,res*sz[v]%mod*inv[n-sz[v]]%mod);
	}
}
int main()
{
	int u,v;
	scanf("%d",&n);
	sum=inv[0]=1;
	for(int i=1;i<=n;i++)
	mul(sum,i),inv[i]=ksm(i,mod-2);
	mul(sum,inv[n]);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs(1,0);
	dfs2(1,0,sum);
	printf("%lld\n",ans);
}