AtCoder Beginner Contest 160题解

前言

本人太菜了,没能AK,写得不好勿喷…
比赛结束30分钟才做出$F$,我太nan了.
同情作者的就点赞资瓷一下吧.

正题

1. 水
2. 水
3. 求一下最大间距即可.
4. $O(n^2)$枚举,两点最短距离=$min(j-i,abs(y-j)+abs(x-i)+1)(j>i)$
5. 贪心.(有一点思维难度).首先,只保留$p$中的前$x$大、$q$中的前$y$大,然后考虑把$r$中的换到这中间,显然把最大的替换最小的最优.
6. 猥琐的换根计数dp.$f[x]=\frac{(sz[x]-1)!}{\prod sz[son]!}\ast \prod f[son],f[x]表示以x为根的子树的总编号方案数(对于叶子节点f=1).$
   $\frac{(sz[x]-1)!}{\prod sz[son]!}$这个是多重集的排列,因为每个孩子是独立的,所以一个孩子$son$子树的全部点看作一个多重集,多重集的贡献为$f[son]$.剩下的就是换根dp啦.(主要就是撤销影响).

$A$

char s[12];
int main() {
	scanf("%s",s+1);
	if(s[3]==s[4]&&s[5]==s[6]) puts("Yes");
	else puts("No");

	return 0;
}

$B$

int main() {
	int n; qr(n);
	int ans=n/500*1000+(n%500)/5*5;
	pr2(ans);
	return 0;
}

$C$

int m,n,a[N],ans;

int main() {
	qr(m);qr(n);
	for(int i=1;i<=n;i++) qr(a[i]),ans=max(ans,a[i]-a[i-1]);
	ans=max(ans,m-a[n]+a[1]);
	pr2(m-ans);
	return 0;
}

$D$

int n,x,y,ans[N];

int main() {
	qr(n); qr(x); qr(y);
	for(int i=1;i<n;i++)
		for(int j=i+1;j<=n;j++) 
			ans[min(j-i,abs(j-y)+abs(i-x)+1)]++;
	for(int i=1;i<n;i++) pr2(ans[i]);
	return 0;
}

$E$

int x,y,a,b,c,A[N],B[N],C[N];
bool cmp(int x,int y) {return x>y;}

int main() {
	qr(x); qr(y); qr(a); qr(b); qr(c);
	for(int i=1;i<=a;i++) qr(A[i]);
	for(int i=1;i<=b;i++) qr(B[i]);
	for(int i=1;i<=c;i++) qr(C[i]);
	sort(A+1,A+a+1,cmp);
	sort(B+1,B+b+1,cmp);
	sort(C+1,C+c+1,cmp);
	A[0]=B[0]=1<<30;
	for(int i=1,j=x,k=y;i<=c;i++) {
		int u=C[i]-A[j],v=C[i]-B[k];
		if(u>=v&&u>0&&j) A[j--]=C[i];
		else if(v>=u&&v>0&&k) B[k--]=C[i];
		else break;
	}
	ll ans=0;
	for(int i=1;i<=x;i++) ans+=A[i];
	for(int i=1;i<=y;i++) ans+=B[i];
	pr2(ans);
	return 0;
}

$F$

#include<map>
#include<set>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#define lc (x<<1)
#define rc (x<<1|1)
#define gc getchar()//(p1==p2&&(p2=(p1=buf)+fread(buf,1,size,stdin),p1==p2)?EOF:*p1++)
#define mk make_pair
#define pi pair<int,int>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int N=2e5+10,size=1<<20,mod=1e9+7;

//char buf[size],*p1=buf,*p2=buf;
template<class o> void qr(o &x) {
	char c=gc; x=0; int f=1;
	while(!isdigit(c)){if(c=='-')f=-1; c=gc;}
	while(isdigit(c)) x=x*10+c-'0',c=gc;
	x*=f;
}
template<class o> void qw(o x) {
	if(x/10) qw(x/10);
	putchar(x%10+'0');
}
template<class o> void pr1(o x) {
	if(x<0)x=-x,putchar('-');
	qw(x); putchar(' ');
}
template<class o> void pr2(o x) {
	if(x<0)x=-x,putchar('-');
	qw(x); puts("");
}

int n;
ll jc[N],inv[N],sz[N],f[N],ans[N],g[N];
struct edge{int y,next;}a[N<<1];int len,last[N];
void ins(int x,int y) {a[++len]=(edge){y,last[x]};last[x]=len;}

void dfs1(int x,int fa) {
	sz[x]=f[x]=1;
	for(int k=last[x],y;k;k=a[k].next)
		if((y=a[k].y)^fa) {
			dfs1(y,x);
			sz[x]+=sz[y];
			f[x]=f[x]*inv[sz[y]]%mod*f[y]%mod;
		}
	f[x]=f[x]*jc[sz[x]-1]%mod;
}

ll power(ll a,ll b) {
	ll c=1;
	while(b) {
		if(b&1) c=c*a%mod;
		a=a*a%mod; b=b>>1;
	}
	return c;
}
			
void dfs2(int x,int fa) {
	ans[x]=f[x]*inv[sz[x]-1]%mod*jc[n-1]%mod*inv[n-sz[x]]%mod*g[fa]%mod;
	for(int k=last[x],y;k;k=a[k].next)
		if((y=a[k].y)^fa) {
			g[x]=f[x]*inv[sz[x]-1]%mod*jc[n-sz[y]-1]%mod*jc[sz[y]]%mod*power(f[y],mod-2)%mod*g[fa]%mod*inv[n-sz[x]]%mod;
			dfs2(y,x);
		}
}

int main() {
	qr(n);for(int i=jc[0]=1;i<=n;i++) jc[i]=jc[i-1]*i%mod;
	inv[n]=power(jc[n],mod-2);
	for(int i=n; i;i--) inv[i-1]=inv[i]*i%mod;
	for(int i=1,x,y;i<n;i++) 
		qr(x),qr(y),ins(x,y),ins(y,x);
	dfs1(1,0);
	g[0]=1;dfs2(1,0);
	for(int i=1;i<=n;i++) pr2(ans[i]);	
	return 0;
}