数字权重
题目
小a有一个n位的数字,但是它忘了各个位上的数是什么,现在请你来确定各个位上的数字,满足以下条件:
设第i位的数为ai,其中a1为最高位,an为最低位,K为给定的数字请你求出满足条件的方案数
输入
两个整数N, K
若存在无解的情况,请输出0
输出
一个整数表示答案,对109 + 7取模
示例1
输入
2 3
输出
6
说明
满足条件的数有:14, 25, 36, 47, 58, 69
示例2
输入
2 -3
输出
7
说明
满足条件的数有:41, 52, 63, 74, 85, 96, 30
示例3
输入
4 3
输出
600
说明
可能的方案有:1234, 1334
示例4
输入
4 -3
输出
700
备注:
对于30%的数据:n, |k| = 5
对于60%的数据:n, |k| ≤ 1000
对于100%的数据:n, |k| ≤ 1013
保证n > 1
1.化简条件2可得,an-a1=K
(下述看他人解析得来)
2.an∈[0,9],a1∈[0,9]
3.K的范围-9<=K<=9
4.|K|>=10,输出0
5.n=2,K>=0时,9-K种(a1作为最高位不能为0)
6.n=2,K<0时,10+K种
7.n=3时,多了中间一位数字,所以在n=2的基础上*10
8.综上,10 ^ (n-2) *(9-K),(9>=K>=0)或者10 ^ (n-2) *(10+K),(-9<=K<0)
#include<iostream>
#include<cmath>
#define int long long
using namespace std;
const int mod=1e9+7;
int power(int a,int n){
int ans=1;
while(n){
if(n%2){
ans=(ans*a)%mod;
}
a=a*a%mod;
n/=2;
}
return ans;
}
signed main(){
int n,k;
cin>>n>>k;
if(fabs(k)>=10)
cout<<0<<endl;
else if(k>=0&&k<=9){
cout<<power(10,n-2)*(9-k)%mod<<endl;
}
else if(k<0&&k>=-9){
cout<<power(10,n-2)*(10+k)%mod<<endl;
}
return 0;
}