Vijos-p1232原子力発電所の問題(動的計画法+再帰)

原子力発電所の問題

説明

原子力発電所には核物質を放出するためのN個のピットがあり、ピットは一直線に並んでいます。核物質をM個の連続したピットに入れると爆発が発生するため、一部のピットに核物質が入れられない場合があります。

ここで、計算してください。与えられたNとMについて、爆発しない核物質を配置するための計画の総数を見つけます。

フォーマット

入力フォーマット

入力ファイルには、1行、2つの正の整数NおよびMしかありません。

出力フォーマット

出力ファイルには、ソリューションの総数を表す正の整数が1つだけあります。

例1

サンプル入力1

4 3 

サンプル出力1

13

制限

1秒

促す

すべてのデータn <= 50、m <= 5

問題解決

オプション1

すべての状況を考慮してください

// dp[i][j] 表示第i个坑已经放置j个连续时的方案数
// 每个坑有放与不放两个选择
// 放: dp[i][j] = dp[i - 1][j - 1]
// 不放(即把当前留为空): dp[i][0] += dp[i - 1][k] (0 <= k < m)

コード

#include <algorithm>
#include <iostream>
#include <string>
using namespace std;
typedef long long ll;
const int maxn = 52;
int N, M;
ll ans;
ll dp[maxn][6];  

int main() {
    
    
    cin >> N >> M;
    dp[1][1] = 1, dp[1][0] = 1;

    for (int i = 2; i <= N; i++)
        for (int j = 0; j < M; j++) {
    
    
            if (j)
                dp[i][j] = dp[i - 1][j - 1];
            else
                for (int k = 0; k < M; k++) dp[i][j] += dp[i - 1][k];
        }
    for (int j = 0; j < M; j++) ans += dp[N][j];
    cout << ans << endl;

    system("pause");
    return 0;
}

オプションII

カテゴリでアイデアについて話し合い、重要なk値を決定します

// dp[i] 表示第i个坑时的方案数
// i = 0时, 没有坑只有一种方案dp[0] = 1
// i < m时, 可以任意放, 每个坑有两种方案, 则 dp[i] = dp[i - 1] * 2;
// i = m时, 除去全部放的方案, 则 dp[i] = dp[i - 1] * 2 - 1;
// i > m时, 要除去更多的方案, 则 dp[i] = dp[i - 1] * 2 - k;
/* 关于k的值 k = dp[i - m - 1]
此时依旧有两种方案放或者不放,总的方案数为DP[i-1]*2,而减去的为连续出现m个的可能情况;
当i > m + 1时可以将连续的m个炸弹看做一个炸弹,但是此时这个炸弹是必须存在的;
所以当i == m+1时方案数为DP[0],(即记录初始状态为)之后的状态类似i<m时。
*/

コード

#include <algorithm>
#include <iostream>
#include <string>
using namespace std;
typedef long long ll;
const int maxn = 52;
int N, M;
ll ans;
ll dp[maxn];

int main() {
    
    
    cin >> N >> M;
    dp[0] = 1;
    for (int i = 1; i <= N; i++) {
    
    
        if (i < M)
            dp[i] = dp[i - 1] * 2;
        else if (i == M)
            dp[i] = dp[i - 1] * 2 - 1;
        else if (i > M)
            dp[i] = dp[i - 1] * 2 - dp[i - M - 1];
    }
    cout << dp[N] << endl;

    system("pause");
    return 0;
}

おすすめ

転載: blog.csdn.net/qq_45349225/article/details/109561202