阿里9.4编程第一题,排列组合
阿里9.4编程第一题,排列组合,只AC了0.9,请各位大佬帮忙看看什么问题
#include <bits/stdc++.h> using namespace std; long long c(int n, int k){ long long tmp = 1; for (int i = 0; i < k; ++i) { tmp *= n-i; tmp /= i+1; } return tmp; } int main() { // ios::sync_with_stdio(false); // cin.tie(nullptr); // cout.tie(nullptr); int n; cin>>n; vector<int> a(4); for (int i = 0; i < 4; ++i) { cin>>a[i]; } sort(a.begin(), a.end()); long long res = 1; long long t, tmp = n*n; for (int i = 0; i < 3; ++i) { if(a[i] > 0){ t = c(tmp, a[i]); if(t > 998244353) t %= 998244353; res *= t; tmp -= a[i]; if(res > 998244353) res %= 998244353; } } cout<<res<<endl; return 0; }