Submission #1029356

#	Time	Username	Problem	Language	Result	Execution time	Memory
1029356		codefox	Skyscraper (JOI16_skyscraper)	C++14	15 / 100	3 ms	1628 KiB

skyscraper

    #include <bits/stdc++.h>

    using namespace std;

    #define int long long
    const int mod = 1e9 + 7;

    int32_t main()
    {
        int n, l;
        cin >> n >> l;
        vector<int> nums(n);
        for (int i = 0; i < n; i++)
        {
            cin >> nums[i];
        }
        sort(nums.begin(), nums.end());
        vector<vector<vector<vector<int>>>> dp(n+1, vector<vector<vector<int>>>(l+1, vector<vector<int>>(n+1, vector<int>(3, 0))));
        dp[1][0][1][0] = 1;
        dp[1][0][1][1] = 2;
        for (int i = 1; i < n; i++)
        {
            for (int j = 0; j <= l; j++)
            {
                for (int k = 1; k < n; k++)
                {
                    for (int m =0; m < 3; m++)
                    {
                        int nsum = j;
                        nsum = j+(k*2-m)*(nums[i]-nums[i-1]);
                        if(nsum > l) continue;
                        if (nsum<0) continue;
                        dp[i+1][nsum][k][m] += dp[i][j][k][m]*(k*2-m);
                        dp[i+1][nsum][k-1][m] += dp[i][j][k][m]*(k-1);
                        dp[i+1][nsum][k+1][m] += dp[i][j][k][m]*(k+1-m);
                        if(m!=2)
                        {
                            dp[i+1][nsum][k+1][m+1] += dp[i][j][k][m]*(2-m);
                            if(k) dp[i+1][nsum][k][m+1] += dp[i][j][k][m]*(2-m);
                            dp[i+1][nsum][k+1][m+1]%=mod;
                            dp[i+1][nsum][k][m+1]%=mod;
                        }
                        dp[i+1][nsum][k][m]%=mod;
                        dp[i+1][nsum][k-1][m]%=mod;
                        dp[i+1][nsum][k+1][m]%=mod;
                    }
                }
            }
        }
        int sum = 0;
        for (int i = 0; i <= l; i++)
        {
            sum += dp[n][i][1][2];
            sum%=mod;
        }
        cout << sum;
        return 0;
    }

• Subtask 1: 0 / 5
• Subtask 2: 15 / 15
• Subtask 3: 0 / 80