AtCoder Beginner Contest 156

传送门

A - Beginner

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int main() {
    //freopen("in.txt","r",stdin);
    int n,r;
    scanf("%d%d",&n,&r);
    if(n>=) printf("%d\n",r);
    else printf("%d\n",r+*(-n));
    return ;
}

A.cpp

B - Digits

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int main() {
    //freopen("in.txt","r",stdin);
    int n,k,ans=;
    scanf("%d%d",&n,&k);
    while(n) {
        n/=k;
        ans++;
    }
    printf("%d\n",ans);
    return ;
}

B.cpp

C - Rally

题意：在一条直线上，有N个人，分别在Xi位置上，要求找一点使得N个人到该点的距离平方和最小，并输出距离平方和。

数据范围：$1 \leq N,Xi \leq 100$

题解：暴力枚举每一个1~100每个位置即可。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=;
int a[N];
int main() {
    //freopen("in.txt","r",stdin);
    int n,ans=1e9;
    scanf("%d",&n);
    for(int i=,x;i<n;i++) {
        scanf("%d",&a[i]);
    }
    for(int i=;i<=;i++) {
        int s=;
        for(int j=;j<n;j++) {
            s+=(a[j]-i)*(a[j]-i);
        }
        ans=min(ans,s);
    }
    printf("%d\n",ans);
    return ;
}

C.cpp

D - Bouquet

题意：有n种不同的花，每种一朵，要求至少取1朵但，不能取a朵或b朵花，求能取得方案数（mod 1e9+7)。

数据范围：$2 \leq n \leq 10^{9},1 \leq a < b \leq min(n,2\times 10^{5})$

题解：总方案数-取a朵-取b朵-不取。即为 $2^{n}-C_{n}^{a}-C_{n}^{b}-1$。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int MD=1e9+;
int quick_pow(int x,int y) {
    int ans=;
    while(y) {
        if(y&) ans=1LL*ans*x%MD;
        y>>=;
        x=1LL*x*x%MD;
    }
    return ans;
}
void add(int &x,int y) {
    x+=y;
    if(x>=MD) x-=MD;
    if(x<) x+=MD;
}
int main() {
    //freopen("in.txt","r",stdin);
    int n,a,b;
    scanf("%d%d%d",&n,&a,&b);
    int ans=quick_pow(,n),s=;
    add(ans,-);
    for(int i=;i<=b;i++) {
        s=1LL*s*(n-i+)%MD,s=1LL*s*quick_pow(i,MD-)%MD;
        if(i==a) add(ans,-s);
        if(i==b) add(ans,-s);
    }
    printf("%d\n",ans);
    return ;
}

D.cpp

E - Roaming

题意：有n个房间，每个房间初始有一个人，每次移动可将一个人移到另一个房间，求k次移动后n个房间的状态数。

数据范围：$3 \leq n \leq 2\times 10^{5},2 \leq k \leq 10^{9}$

题解:考虑房间人数为0的数量。

1.没有房间的人数为0的移法：把1移到2，然后2~3来回移动k-2次，最后在移回1。

2.有且只有一个房间的人数为0的移法：把1移到其它房间，然后在其它房间任意移动。

3.超过一个房间的人数为0的移法：结合上面两种移法即可。

枚举0的个数i，然后就是将i个人分到n-i的房间的子问题。

即为：$\sum_{0}^{min(k,n-1)}C_{n}^{i}*C_{n-1}^{i}$

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=4e5+;
const int MD=1e9+;
int f[N],inv[N];
int quick_pow(int x,int y) {
    int ans=;
    while(y) {
        if(y&) ans=1LL*ans*x%MD;
        y>>=;
        x=1LL*x*x%MD;
    }
    return ans;
}
int C(int n,int m) {
    return 1LL*f[n]*inv[m]%MD*inv[n-m]%MD;
}
void add(int &x,int y) {
    x+=y;
    if(x>=MD) x-=MD;
}
void init() {
    f[]=;
    for(int i=;i<N;i++) f[i]=1LL*f[i-]*i%MD;
    inv[N-]=quick_pow(f[N-],MD-);
    for(int i=N-;i>=;i--) inv[i]=1LL*inv[i+]*(i+)%MD;
}
int main() {
    //freopen("in.txt","r",stdin);
    init();
    int n,k,ans=;
    scanf("%d%d",&n,&k);
    k=min(k,n-);
    for(int i=;i<=k;i++) {
        add(ans,1LL*C(n,i)*C(n-,i)%MD);
    }
    printf("%d\n",ans);
    return ;
}

E.cpp

F - Modularness

题意：有k个非负整数di，q个查询，每次查询给三个整数n，x，m，构造出一个长度为n的序列，其中:

$a_{0}=x,a_{i}=a_{i-1}+d_{(i-1) \bmod k}$，输出存在多少个下标i（0<=i<n-1）满足$a_{i} \bmod m <a_{i+1}\bmod m$。

数据范围：$1 \leq k,q \leq 5000,0 \leq di,xi \leq 10^{9},2 \leq ni,mi \leq 10^{9}$

题解：总数 - $a_{i} \bmod m =a_{i+1}\bmod m$的数目 - $a_{i} \bmod m >a_{i+1}\bmod m$的数目。

可以先将di对m取模，这样di是一个小于m的数。

1.$a_{i} \bmod m =a_{i+1}\bmod m：di=0$

2.$a_{i} \bmod m >a_{i+1}\bmod m：\left \lfloor \frac{a_{i}}{m} \right \rfloor=\left \lfloor \frac{a_{i+1}}{m} \right \rfloor-1$

1可以很轻易的算出，2等价于$\left \lfloor \frac{a_{n-1}}{m} \right \rfloor - \left \lfloor \frac{a_{0}}{m} \right \rfloor$。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=5e3+;
int d[N],t[N],k,q;
ll cal(int n,int x,int m) {
    ll s=;
    for(int i=;i<k;i++) {
        t[i]=d[i]%m;
        if(t[i]==) t[i]=m;
        s+=t[i];
    }
    ll ans=x+s*((n-)/k);
    for(int i=;i<(n-)%k;i++) {
        ans+=t[i];
    }
    return n--(ans/m-x/m);
}
int main() {
    //freopen("in.txt","r",stdin);
    scanf("%d%d",&k,&q);
    for(int i=;i<k;i++) {
        scanf("%d",&d[i]);
    }
    for(int i=,n,x,m;i<q;i++) {
        scanf("%d%d%d",&n,&x,&m);
        printf("%lld\n",cal(n,x,m));
    }
    return ;
}

F.cpp