Atcodr abc 172题解(A~E)

A
模拟即可
AC代码：

#include<iostream>
#include<cstdio>
#include <stdio.h>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<map>
#include<vector>
#include <set>
#define ll long long
using namespace std;
int main()
{
    int a;
    cin>>a;
    cout<<a+a*a+a*a*a<<endl;

    return 0;
}

B
直接比较模拟即可
AC代码：

#include<iostream>
#include<cstdio>
#include <stdio.h>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<map>
#include<vector>
#include <set>
#define ll long long
using namespace std;

int main()
{
    char a[200050];
    char c[200050];
    cin>>a;
    cin>>c;
    int sum=0;
    int n=strlen(a);
    for(int i=0;i<n;i++)
    {
        if(a[i]!=c[i])
        {
            sum++;
        }
    }
    cout<<sum<<endl;
    return 0;
}

C
前缀和双指针模拟即可
AC代码：

#include<iostream>
#include<cstdio>
#include <stdio.h>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<map>
#include<vector>
#include <set>
#define ll long long
using namespace std;
ll a[200050];
ll b[200050];
int ans;
int main()
{
    int n,m,k,t;
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        a[i]+=a[i-1];
    }
    for(int i=1;i<=m;i++)
    {
        cin>>b[i];
        b[i]+=b[i-1];
    }
    for(int i=0,j=m;i<=n;i++)
    {
        if(a[i]<=k)
        {
            while(a[i]+b[j]>k)j--;
            ans=max(ans,i+j);
        }
    }
    cout<<ans<<endl;
    return 0;
}

D
直接nlogn暴力即可。
AC代码：

#include<iostream>
#include<cstdio>
#include <stdio.h>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<map>
#include<vector>
#include <set>
#define ll long long
using namespace std;
const int N=1e7+5;
int a[N];
ll ans=0;
int main()
{
    for(int i=1;i<=10000000;i++)
    {
        for(int j=i;j<=10000000;j+=i)//（N*log（N）级别的打表）
        {
            a[j]++;
        }
    }
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
        ans=ans+(long long)i*a[i];//（注意开long long）
    cout<<ans<<endl;
    return 0;
}

E
固定A数组有A(N,M)全排列，M个中取N个全排列
考虑B数组M个数中取出N个数，然后使用容斥原理
求出第I个数不为I
第i个数为i的并集和= 一个数不为i - 两个数不为i + 三个数不为i -……
最后乘以A的全排列即可

#include<iostream>
#include<cstdio>
#include <stdio.h>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<map>
#include<vector>
#include <set>
#define ll long long
#define IO ios::sync_with_stdio(false);cin.tie();cout.tie(0)
using namespace std;
const ll mod=1e9+7;
const ll N=5e5+5;
ll n,m;
ll fac[N],infac[N];
ll powm(ll a,ll b,ll mod)//快速幂
{
    ll ans=1;
    while(b)
    {
        if(b&1)
        {
            ans=ans*a%mod;
        }
        b=b>>1;
        a=a*a%mod;
    }
    return ans;
}
void init()//初始化
{
    fac[0]=infac[0]=1;
    for(int i=1; i<=m; i++)
    {
        fac[i]=fac[i-1]*i%mod;
        infac[i]=powm(fac[i],mod-2,mod);
    }
}
int main()
{
    IO;
    cin>>n>>m;
    init();
    ll res=0;
    for(int i=0;i<=n;i++)//容斥原理计算B
    {
        if(i&1)
        {
            res=(res-fac[n]*infac[i]%mod*infac[n-i]%mod*fac[m-i]%mod*infac[m-n]%mod+mod)%mod;
        }
        else
        {
            res=(res+fac[n]*infac[i]%mod*infac[n-i]%mod*fac[m-i]%mod*infac[m-n]%mod)%mod;
        }
    }
    for(int i=m-n+1;i<=m;i++) res=res*i%mod;//A的全排列
    cout<<(res+mod)%mod<<endl;
    return 0;
}