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}^{cnt1}cnt1!C_{n-x}^{cnt2}cnt2!(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$ 个居民。强盗从 1 号广场开始抓人，居民任意选择一条联通的道路跑。假如居民和强盗都按最优的方式跑，强盗最后能够抓到几个人。

思路：维护三个值，当前子树的总人数 sum，当前子树叶节点个数 cnt，当前子树的人数的最大值 mx。

• 当 $\frac{sum}{cnt}+(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$ ，问对所有子区间的 mex 取 mex 的结果。$(1\le a_i\le n,1\le n\le 10^5)$

思路：以一种动态更新的思路来看。

• 假设当前询问的是 mex 是否等于 $a[i]$ ，那么只需要在 $last[a[i]]+1$ 和 $i-1$ 的位置之间出现过 $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;
}