QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#232917#6410. Classical DP Problemiulia_morariuWA 0ms3804kbC++141.8kb2023-10-31 00:47:122023-10-31 00:47:13

Judging History

你现在查看的是最新测评结果

  • [2023-10-31 00:47:13]
  • 评测
  • 测评结果:WA
  • 用时:0ms
  • 内存:3804kb
  • [2023-10-31 00:47:12]
  • 提交

answer

#include <bits/stdc++.h>
#define Mod 1000000007

using namespace std;

int k;
int din(vector <int> a){
    int n = a.size();
    int x = 0;
    int it = n - 1;
    if(n == k) x = 0;
    else x = a[k];

    int dp[n + 1][n + 1];
    for(int i = 0; i <= n; i++){
        for(int j = 0; j <= n; j++) dp[i][j] = 0;
    }

    dp[0][ x ] = 1;

    for(int i = 0; i < n; i++){
        for(int j = 0; j <= x; j++){
            // acoperim o cloana neimportanta
            dp[i + 1][j] += (a[i] - j) * dp[i][j];
            dp[i + 1][j] %= Mod;
            // acoperim o coloana importanta
            if(j > 0){
                dp[i + 1][j - 1] += j * dp[i][j];
                dp[i + 1][j - 1] %= Mod;
            }
        }
    }


    return dp[k][0];
}

int main(){
    cin.tie(0);ios::sync_with_stdio(0);

    int n; cin >> n;
    vector <int> v(n);
    for(int i = 0; i < n; i++) cin >> v[i];
    reverse(v.begin(), v.end());
    k = 0;

    for(int i = 0; i < n; i++){
        bool ok = 1;
        for(int j = 0; j <= i; j++){
            if(v[j] < i + 1){
                ok = 0;
                break;
            }
        }
        if(!ok){
            k = i;
            break;
        }
    }

    if(n == 1) k = 1;

    cout << k << " ";


    int val1 = din(v);
    //cout << "val1 = "  << val1 << endl;

    vector<int> v1(n);
    for(int i = 0; i < n; i++) v1[i] = 0;
    for(int i = 0; i < n; i++){
        for(int j = 0; j < v[i]; j++){
            v1[j]++;
        }
    }
    //reverse(v1.begin(), v1.end());

    int val2 = din(v1);
    //cout << "val2 = " << val2 << endl;

    int fact = 1;
    for(int i = 2; i <= k; i++){
        fact *= k;
        fact %= Mod;
    }

    cout << val1 + val2 - fact << endl;



    return 0;
}

详细

Test #1:

score: 100
Accepted
time: 0ms
memory: 3608kb

input:

3
1 2 3

output:

2 6

result:

ok 2 number(s): "2 6"

Test #2:

score: 0
Accepted
time: 0ms
memory: 3804kb

input:

1
1

output:

1 1

result:

ok 2 number(s): "1 1"

Test #3:

score: 0
Accepted
time: 0ms
memory: 3612kb

input:

2
1 1

output:

1 2

result:

ok 2 number(s): "1 2"

Test #4:

score: -100
Wrong Answer
time: 0ms
memory: 3620kb

input:

2
2 2

output:

0 -1

result:

wrong answer 1st numbers differ - expected: '2', found: '0'