这样也能过

#include <bits/stdc++.h>
using namespace std;
int n,k,ans,a[10];
int main(){
	cin>>n>>k;
	if(k==2){
		for(int i=1;i<=n;i++) if(n-i>0)ans++;
		ans/=2;
	}else if(k==3){
		for(int i=1;i<=n;i++) for(int j=n-i;j>=i;j--) if(n-i-j>0&&n-i-j>=j)ans++;
	}else if(k==4){
		for(int i=1;i<=n;i++) for(int j=n-i;j>=i;j--) for(int k=n-i-j;k>=j;k--) if(n-i-j-k>0&&n-i-j-k>=k)ans++;
	}
	else if(k==5){
		for(int i=1;i<=n;i++)
			for(int j=n-i;j>=i;j--)
				for(int k=n-i-j;k>=j;k--)
					for(int a=n-i-j-k;a>=k;a--)
						if(n-i-j-k-a>0&&n-i-j-k-a>=a)ans++;	
	}else{
		for(int i=1;i<=n;i++)
			for(int j=n-i;j>=i;j--)
				for(int k=n-i-j;k>=j;k--)
					for(int a=n-i-j-k;a>=k;a--)
						for(int b=n-i-j-k-a;b>=a;b--)
							if(n-i-j-k-a-b>0&&n-i-j-k-a-b>=b)ans++;	
	}
	cout<<ans<<endl;
    return 0;
}