AtCoder Beginner Contest 132

C - Divide the Problems

题意：问有多少种不同的k使得难度小于k和大于等于k的题目数量相等。

思路：如果是奇数的话不可分，偶数的话排序，a[n/2]-a[n/2-1]即是答案。

#include<bits/stdc++.h>
using namespace std;
int a[100005];
int n;
int main(){
	cin>>n;
	for(int i=0;i<n;i++){
		scanf("%d",&a[i]);
	}
	sort(a,a+n);
	int ans=0;
	if(n%2==0){
		ans=a[n/2]-a[n/2-1];
	}
	printf("%d",ans);
}

D - Blue and Red Balls

题意：有n个球，k个蓝球，n-k个红球，输出k个数其中i（1<=i<=k)表示可以i次收集完蓝球。

思路：排列组合题。红球的放法c(n-k+1,i)，蓝球的放法c(k-1,i-1)。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll fac[2005];
const int mod=1e9+7;
ll inv(ll b){return b==1?1:(mod-mod/b)*inv(mod%b)%mod;}
ll c(int n,int r){
	if(r==0||r==n)
		return 1;
	if(n<r)
		return 0;
	return fac[n]*inv(fac[r])%mod*inv(fac[n-r])%mod;
}
int main(){
	fac[1]=1;
	fac[0]=1;
	for(int i=2;i<=2000;i++)
		fac[i]=fac[i-1]*i%mod;
	int n,k;
	scanf("%d%d",&n,&k);
	ll ans=1;
	for(int i=1;i<=k;i++){
		ll r,b;
		r=c(n-k+1,i);
		b=c(k-1,i-1);
		ans=r*b%mod;
		printf("%lld\n",ans);
	}
	return 0;
}

E - Hopscotch Addict

题意：一个有向图，给定起点和终点，每次只能走三步，不限制走的次数。问最终能否在第三步到达终点。

思路：暴力搜索即可。

#include<bits/stdc++.h>
using namespace std;
vector<int>g[100005];
int dis[100005][3];
int main(){
	int n,m,s,t,u,v;
	memset(dis,-1,sizeof(dis));
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++){
		scanf("%d%d",&u,&v);
		g[u].push_back(v);
	}
	scanf("%d%d",&s,&t);
	queue<pair<int,int> >q;
	q.push({s,0});
	dis[s][0]=0;
	while(!q.empty()){
		int v=q.front().first;
		int l=q.front().second;
		q.pop();
		for(int i=0;i<g[v].size();i++){
			int ll=(l+1)%3;
			if(dis[g[v][i]][ll]>=0)
				continue;
			dis[g[v][i]][ll]=dis[v][l]+1;
			q.push({g[v][i],ll});
		}
	}
	if(dis[t][0]>=0)
		dis[t][0]/=3;
	printf("%d\n",dis[t][0]);
}