Submission #112570

#	Time	Username	Problem	Language	Result	Execution time	Memory
112570		Kastanda	Skyscraper (JOI16_skyscraper)	C++11	100 / 100	382 ms	3932 KiB

skyscraper

// I am.
#include<bits/stdc++.h>
using namespace std;
const int N = 109, L = 1009, Mod = 1e9 + 7;
int n, l, A[N], dp[N][L][2][2], dpt[N][L][2][2];
inline void Add(int j, int k, int s, int e, int val)
{
    dp[j][k][s][e] += val;
    if (dp[j][k][s][e] >= Mod)
        dp[j][k][s][e] -= Mod;
}
int main()
{
    scanf("%d%d", &n, &l);
    for (int i = 0; i < n; i ++)
        scanf("%d", &A[i]);
    sort(A, A + n);
    dp[1][0][0][0] = dp[1][0][0][1] = 1;
    dp[1][0][1][0] = dp[1][0][1][1] = 1;

    for (int i = 0; i < n - 1; i ++)
    {
        memcpy(dpt, dp, sizeof(dp));
        memset(dp, 0, sizeof(dp));
        for (int j = 1; j <= i + 1; j ++)
            for (int k = 0; k <= l; k ++)
                for (int s = 0; s <= 1; s ++)
                    for (int e = 0; e <= 1; e ++)
                    {
                        int _k = k + (j + j - s - e) * (A[i + 1] - A[i]);
                        if (_k > l)
                            continue;
                        Add(j - 1, _k, s, e, dpt[j][k][s][e] * 1LL * (j - 1) % Mod);
                        Add(j + 1, _k, s, e, dpt[j][k][s][e] * 1LL * (j - 1) % Mod);
                        Add(j, _k, s, e, dpt[j][k][s][e] * 2LL * (j - 1) % Mod);
                        if (!s)
                        {
                            Add(j, _k, 1, e, dpt[j][k][s][e]);
                            Add(j + 1, _k, 1, e, dpt[j][k][s][e]);
                            Add(j, _k, 0, e, dpt[j][k][s][e]);
                            Add(j + 1, _k, 0, e, dpt[j][k][s][e]);
                        }
                        if (!e)
                        {
                            Add(j, _k, s, 1, dpt[j][k][s][e]);
                            Add(j + 1, _k, s, 1, dpt[j][k][s][e]);
                            Add(j, _k, s, 0, dpt[j][k][s][e]);
                            Add(j + 1, _k, s, 0, dpt[j][k][s][e]);
                        }
                    }
    }
    int tot = 0;
    for (int i = 0; i <= l; i ++)
        tot = (tot + dp[1][i][1][1]) % Mod;
    return !printf("%d\n", tot);
}

Compilation message (stderr)

skyscraper.cpp: In function 'int main()':
skyscraper.cpp:14:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d", &n, &l);
     ~~~~~^~~~~~~~~~~~~~~~
skyscraper.cpp:16:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d", &A[i]);
         ~~~~~^~~~~~~~~~~~~

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