蓝桥杯 算法训练 ALGO-3 K好数

#include<iostream>
#include<algorithm>
#define M 1000000007
using namespace std;
int main()
{
	long long k,l;
	long long  i,j,r;
	long long  f[200][200];
	while(scanf("%lld%lld",&k,&l)!=EOF)
	{
		if(l==1)
		{
			printf("%lld\n",k-1);
			continue;
		}
		
		for(i=0;i<=k-1;i++)
		{
			f[1][i]=1;
		}
		long long cnt;
		for(i=2;i<=l-1;i++)
		{
			for(j=0;j<=k-1;j++)
			{
				cnt=0;
				for(r=0;r<=k-1;r++)
				{
					if(r==j+1||r==j-1)
					continue;
					cnt=(cnt+f[i-1][r])%M;
				}
				
				f[i][j]=cnt;
			}
		}
		long long sum=0;
		for(j=1;j<=k-1;j++)
		{
			//cnt=0;
			for(r=0;r<=k-1;r++)
			{
				if(r==j+1||r==j-1)
				continue;
				sum=(sum+f[l-1][r])%M;
			}
			//sum=(sum+cnt)%M;
		}
		
		printf("%lld\n",sum);
	}
	return 0;
}