ATcoder D - Handstand 2

题目大意：

给一个数N，在小于N的所有数中，找到(A,B)的数量，其中A的第一个数字要等于B的最后的一个数字，A的最后一个数字要等于B的第一个数字。

题解：对从1到N的所有数x，用一个二维数组保存dp[a][b]其中a是x的第一个数字，b是x的最后一个数字。答案就是dp[a][b]*dp[b][a]。。嗯嗯~~~秒

#include<bits/stdc++.h>

using namespace std;

typedef long long ll;

const ll N=;

ll dp[N][N];

ll find(ll x){

    while(x){

        if(x/==){

            return x;

        }

        x/=;

    }

}

int main()

{

    ll n;

    cin>>n;

    for(ll i=;i<=n;i++){

        ll h=i;

        ll a1=h%;

        ll a2=find(h);

        dp[a2][a1]++;

    }

    ll ans=;

    for(ll i=;i<=;i++){

        for(ll j=;j<=;j++){

            ans+=dp[i][j]*dp[j][i];

        }

    }

    cout<<ans<<endl;

    return ;

}

巴特西

ATcoder D - Handstand 2

最新文章

热门文章