สมมติว่าเรามีจำนวนเต็ม n และ k สองจำนวน เราต้องหาจำนวนอาร์เรย์ที่ประกอบด้วยตัวเลขตั้งแต่ 1 ถึง n จำนวนเท่าใดจึงจะมีคู่ผกผัน k คู่พอดี คู่ผกผันสำหรับองค์ประกอบ ith และ jth ในอาร์เรย์ ถ้า i
ดังนั้นหากอินพุตเป็น n =3 และ k =1 เอาต์พุตจะเป็น 2 เนื่องจากอาร์เรย์ [1,3,2] และ [2,1,3] จะมีคู่ผกผันเพียงคู่เดียว
เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -
- กำหนดขนาดอาร์เรย์ 2 มิติหนึ่ง dp (n + 1) x (k + 1)
- dp[0, 0] :=1
- สำหรับการเริ่มต้น i :=1 เมื่อ i<=n อัปเดต (เพิ่ม i โดย 1) ทำ −
- dp[i, 0] :=1
- สำหรับการเริ่มต้น j :=1 เมื่อ j <=k อัปเดต (เพิ่ม j ทีละ 1) ทำ −
- dp[i, j] :=dp[i, j - 1] + dp[i – 1, j]
- dp[i, j] :=dp[i, j] mod ม
- ถ้า j>=i แล้ว −
- dp[i, j] :=(dp[i, j] - dp[i – 1, j - i] + m) mod ม.
- ส่งคืน dp[n, k]
ให้เราดูการใช้งานต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น -
ตัวอย่าง
#include <bits/stdc++.h>
using namespace std;
const int m = 1e9 + 7;
class Solution {
public:
int kInversePairs(int n, int k) {
vector < vector <int> > dp(n + 1, vector <int>(k + 1));
dp[0][0] = 1;
for(int i = 1; i <= n; i++){
dp[i][0] = 1;
for(int j = 1; j <= k; j++){
dp[i][j] = dp[i][j - 1] + dp[i - 1][j];
dp[i][j] %= m;
if(j >= i){
dp[i][j] = (dp[i][j] - dp[i - 1][j - i] + m) % m;
}
}
}
return dp[n][k];
}
};
main(){
Solution ob;
cout << (ob.kInversePairs(4,2));
} อินพุต
4 2
ผลลัพธ์
5