DuongNhi99

A - Rooks (24-11)

Dec 10th, 2020
110
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.86 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. #define ll long long
  3. using namespace std;
  4.  
  5. ll calcNcR(int n, int r) {
  6.     ll p = 1, k = 1;
  7.  
  8.     if(n - r < r) r = n - r;
  9.     if(r != 0)
  10.  
  11.         while(r) {
  12.             p *= n; k *= r;
  13.  
  14.             long long m = __gcd(p, k);
  15.             p /= m; k /= m;
  16.             n--; r--;
  17.         }
  18.     else p = 1;
  19.  
  20.     return p;
  21. }
  22.  
  23. void solve(int test) {
  24.     int n, k; cin >> n >> k;
  25.  
  26.     ll ans = 0;
  27.     if(n < k) cout << "Case " << test << ": " << ans << "\n";
  28.     else {
  29.         ans = calcNcR(n, k) * calcNcR(n, k);
  30.  
  31.         //k giai thua
  32.         for(ll i = 1; i <= k; ++i)
  33.             ans *= i;
  34.  
  35.         cout << "Case " << test << ": " << ans << "\n";
  36.     }
  37.     return;
  38. }
  39.  
  40. int main() {
  41.     //freopen("A-Rooks.inp", "r", stdin);
  42.     //freopen("A-Rooks.out", "w", stdout);
  43.     ios_base::sync_with_stdio(false);
  44.     cin.tie(NULL); cout.tie(NULL);
  45.  
  46.     int t; cin >> t;
  47.     for(int i = 1; i <= t; i++) {
  48.         solve(i);
  49.     }
  50.  
  51.     return 0;
  52. }
  53.  
Advertisement
Add Comment
Please, Sign In to add comment