Codeforces Round #678 Div. 2 Oct/24/2020 22:05UTC+8
比赛链接:https://codeforces.com/contest/1436
比赛记录:https://blog.csdn.net/cheng__yu_/article/details/105395197
C. Binary Search
题意:给定一个二分查找算法,问有多少个 n 的排列通过算法能够找到第 pos 位,且值为 x 。
思路:首先根据二分算法跑,找到 pos 左边小于 x 的数量 cnt1,找到 pos 右边大于 x 的数量 cnt2
- 然后答案就是一个组合数: C x − 1 c n t 1 c n t 1 ! C n − x c n t 2 c n t 2 ! ( n − c n t 1 − c n t 2 − 1 ) ! C_{x-1}^{cnt1}cnt1!C_{n-x}^{cnt2}cnt2!(n-cnt1-cnt2-1)! Cx−1cnt1cnt1!Cn−xcnt2cnt2!(n−cnt1−cnt2−1)!
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e5+5,mod=1e9+7;
int qpow(int b,int n,int mod)
{
int res=1;
while(n)
{
if(n&1) res=1ll*res*b%mod;
b=1ll*b*b%mod;
n>>=1;
}
return res;
}
const int N=10000;
int fac[N+10],finv[N+10];
void init()
{
fac[0]=fac[1]=1;
for(int i=2; i<=N; ++i)
fac[i]=1ll*fac[i-1]*i%mod;
finv[N]=qpow(fac[N],mod-2,mod);
for(int i=N-1; i>=0; --i)
finv[i]=1ll*finv[i+1]*(i+1)%mod;
}
int C(int n,int m)
{
return 1ll*fac[n]*finv[m]%mod*finv[n-m]%mod;
}
int n,x,pos;
int main()
{
init();
scanf("%d%d%d",&n,&x,&pos);
int L=0,R=n;
int cnt1=0,cnt2=0;
while(L<R)
{
int mid=(L+R)>>1;
if(mid<pos) cnt1++,L=mid+1;
else if(mid==pos) L=mid+1;
else if(mid>pos) cnt2++,R=mid;
}
ll ans=1ll*fac[cnt1]*C(x-1,cnt1)%mod*C(n-x,cnt2)%mod*fac[cnt2]%mod*fac[n-cnt1-cnt2-1]%mod;
printf("%lld\n",ans);
return 0;
}
D. Bandit in a City
题意:有 n 个广场,有 n -1 条道路单向联通,且以 1 为根。一开始每个广场都有 a i a_i ai 个居民。强盗从 1 号广场开始抓人,居民任意选择一条联通的道路跑。假如居民和强盗都按最优的方式跑,强盗最后能够抓到几个人。
思路:维护三个值,当前子树的总人数 sum,当前子树叶节点个数 cnt,当前子树的人数的最大值 mx。
- 当 s u m c n t + ( s u m % c n t ? 1 : 0 ) > m x \frac{sum}{cnt}+(sum\%cnt?1:0)>mx cntsum+(sum%cnt?1:0)>mx 时,更新 mx 即可
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=2e5+5;
int n;
vector<int> e[maxn];
int val[maxn];
ll ans;
struct Node
{
ll sum,cnt,mx;
};
Node dfs(int u)
{
if(e[u].empty()) return {val[u],1,val[u]};
ll sum=val[u],cnt=0,mx=0;
for(auto v: e[u])
{
auto res=dfs(v);
sum+=res.sum;
cnt+=res.cnt;
mx=max(mx,res.mx);
}
ll tmp=sum/cnt+(sum%cnt?1:0);
if(tmp>mx) mx=tmp;
ans=max(ans,mx);
return {sum,cnt,mx};
}
int main()
{
scanf("%d",&n);
for(int v=2; v<=n; ++v)
{
int u;
scanf("%d",&u);
e[u].push_back(v);
}
for(int i=1; i<=n; ++i) scanf("%d",&val[i]);
dfs(1);
printf("%lld\n",ans);
return 0;
}
E. Complicated Computations
题意:给定 n 个数的数组 a i a_i ai ,问对所有子区间的 mex 取 mex 的结果。 ( 1 ≤ a i ≤ n , 1 ≤ n ≤ 1 0 5 ) (1\le a_i \le n ,1\le n \le 10^5) (1≤ai≤n,1≤n≤105)
思路:以一种动态更新的思路来看。
- 假设当前询问的是 mex 是否等于 a [ i ] a[i] a[i] ,那么只需要在 l a s t [ a [ i ] ] + 1 last[a[i]]+1 last[a[i]]+1 和 i − 1 i -1 i−1 的位置之间出现过 1 1 1 到 a [ i ] − 1 a[i]-1 a[i]−1 这些数
- 那么可以维护 每一个数出现位置的最小值,如果查询结果大于 last[a[i]] ,就表示 a[i] 是可以得到的
#include <bits/stdc++.h>
#define ls (rt<<1)
#define rs (rt<<1|1)
#define ll long long
using namespace std;
const int maxn=1e5+5;
int n,a[maxn];
int visit[maxn],st[maxn<<2],last[maxn];
void update(int rt,int pos,int L,int R,int val)
{
if(L==R)
{
st[rt]=val;
return;
}
int mid=(L+R)>>1;
if(pos<=mid) update(ls,pos,L,mid,val);
if(pos>mid) update(rs,pos,mid+1,R,val);
st[rt]=min(st[ls],st[rs]);
}
int query(int rt,int l,int r,int L,int R)
{
if(l<=L&&R<=r) return st[rt];
int mid=(L+R)>>1;
int ans=1e9;
if(l<=mid) ans=min(ans,query(ls,l,r,L,mid));
if(r>mid) ans=min(ans,query(rs,l,r,mid+1,R));
return ans;
}
int main()
{
scanf("%d",&n);
bool ok=0;
for(int i=1; i<=n; ++i)
{
scanf("%d",&a[i]);
if(a[i]>1) ok=1;
if(a[i]>1&&query(1,1,a[i]-1,1,n)>last[a[i]]) visit[a[i]]=1;
last[a[i]]=i;
update(1,a[i],1,n,i);
}
if(ok) visit[1]=1;
for(int i=2; i<=n+1; ++i)
if(i>1&&query(1,1,i-1,1,n)>last[i]) visit[i]=1;
int ans=1;
while(visit[ans]) ans++;
printf("%d\n",ans);
return 0;
}