C++ প্রোগ্রামে আমরা কত উপায়ে দুটি শর্তে ব্লক পেইন্ট করতে পারি তা গণনা করে

ধরুন আমাদের তিনটি সংখ্যা N, M এবং K আছে। বিবেচনা করুন যে N ব্লক আছে সেগুলো সারিবদ্ধভাবে সাজানো আছে। আমরা তাদের আঁকা দুটি উপায় অনুসরণ বিবেচনা. দুটি ব্লকের রং আলাদা হয় যদি এবং শুধুমাত্র যদি ব্লকগুলিকে দুটি উপায়ে ভিন্ন রঙে আঁকা হয় -

• প্রতিটি ব্লকের জন্য, এটি আঁকার জন্য M রংগুলির একটি ব্যবহার করুন। (সব রং ব্যবহার করা বাধ্যতামূলক নয়)

• একই রঙে আঁকা সংলগ্ন ব্লকগুলির সর্বাধিক K জোড়া থাকতে পারে৷

উত্তরটি খুব বড় হলে, ফলাফল মোড 998244353 ফেরত দিন।

সুতরাং, যদি ইনপুটটি N =3 এর মত হয়; M =2; K =1, তাহলে আউটপুট হবে 6, কারণ আমরা সেগুলোকে 112, 121, 122, 211, 212, এবং 221 এই বিভিন্ন ফরম্যাটে পেইন্ট করতে পারি।

পদক্ষেপ

এটি সমাধান করতে, আমরা এই পদক্ষেপগুলি অনুসরণ করব -

maxm := 2^6 + 5
p := 998244353
Define two large arrays fac and inv or size maxm
Define a function ppow(), this will take a, b, p,
ans := 1 mod p
a := a mod p
while b is non-zero, do:
   if b is odd, then:
      ans := ans * a mod p
   a := a * a mod p
   b := b/2
return ans
Define a function C(), this will take n, m,
if m < 0 or m > n, then:
   return 0
return fac[n] * inv[m] mod p * inv[n - m] mod p
From the main method, do the following
fac[0] := 1
for initialize i := 1, when i < maxm, update (increase i by 1), do:
   fac[i] := fac[i - 1] * i mod p
inv[maxm - 1] := ppow(fac[maxm - 1], p - 2, p)
for initialize i := maxm - 2, when i >= 0, update (decrease i by 1), do:
   inv[i] := (i + 1) * inv[i + 1] mod p
ans := 0
for initialize i := 0, when i <= k, update (increase i by 1), do:
   t := C(n - 1, i)
   tt := m * ppow(m - 1, n - i - 1, p)
   ans := (ans + t * tt mod p) mod p
return ans

উদাহরণ

আরো ভালোভাবে বোঝার জন্য আসুন নিচের বাস্তবায়ন দেখি -

#include <bits/stdc++.h>
using namespace std;

const long maxm = 2e6 + 5;
const long p = 998244353;
long fac[maxm], inv[maxm];

long ppow(long a, long b, long p){
   long ans = 1 % p;
   a %= p;
   while (b){
      if (b & 1)
         ans = ans * a % p;
      a = a * a % p;
      b >>= 1;
   }
   return ans;
}
long C(long n, long m){
   if (m < 0 || m > n)
      return 0;
   return fac[n] * inv[m] % p * inv[n - m] % p;
}
long solve(long n, long m, long k){
   fac[0] = 1;
   for (long i = 1; i < maxm; i++)
      fac[i] = fac[i - 1] * i % p;
   inv[maxm - 1] = ppow(fac[maxm - 1], p - 2, p);
   for (long i = maxm - 2; i >= 0; i--)
      inv[i] = (i + 1) * inv[i + 1] % p;
   long ans = 0;
   for (long i = 0; i <= k; i++){
      long t = C(n - 1, i);
      long tt = m * ppow(m - 1, n - i - 1, p) % p;
      ans = (ans + t * tt % p) % p;
   }
   return ans;
}
int main(){
   int N = 3;
   int M = 2;
   int K = 1;
   cout << solve(N, M, K) << endl;
}

ইনপুট

3, 2, 1

আউটপুট

6