提交时间:2023-08-10 20:37:11
运行 ID: 81843
#include<bits/stdc++.h> using namespace std; using i64=long long; i64 n_choose_2(i64 n){ return n*(n-1)/2; } i64 n_choose_4(i64 n){ return n*(n-1)/2*(n-2)/3*(n-3)/4; } int main(){ int t,n; cin>>t>>n; if(t==1){ cout<<n_choose_2(n); } else if(t==2){ if(n<4){ cout<<0; } else{ cout<<n_choose_4(n); } } if(t==3){ cout<<1+n_choose_2(n)+n_choose_4(n); } return 0; }