oj.uz

답안 #1079213

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1079213 2024-08-28T12:05:03 Z boris_mihov Skyscraper (JOI16_skyscraper) C++17
20 / 100
 5 ms 2140 KB 
#include <algorithm>
#include <iostream>
#include <numeric>
#include <cassert>
#include <vector>

typedef long long llong;
const int MAXN = 15 + 10;
const int MAXL = 1000 + 10;
const int MOD  = 1e9 + 7;

int n, l;
int a[MAXN];
int dp[MAXN][MAXN][2 * MAXN * MAXL][2][2];
bool bl[MAXN][MAXN][2 * MAXN * MAXL][2][2];

int f(int idx, int count, int sum, bool foundB, bool foundE)
{
    if (idx == n + 1)
    {
        if (!foundB || !foundE) return 0;
        return (count == 1 && sum <= MAXN * MAXL + l);
    }

    if (bl[idx][count][sum][foundB][foundE])
    {
        return dp[idx][count][sum][foundB][foundE];
    }

    bl[idx][count][sum][foundB][foundE] = 1;
    llong res = f(idx + 1, count + 1, sum - 2 * a[idx], foundB, foundE);
    if (!foundB)
    {
        res += f(idx + 1, count + 1, sum - a[idx], 1, foundE);
    }

    if (!foundE)
    {
        res += f(idx + 1, count + 1, sum - a[idx], foundB, 1);
    }

    if (count >= 1) 
    {
        res += (2LL * count - foundB - foundE) * f(idx + 1, count, sum, foundB, foundE);
        if (!foundB)
        {
            res += (count - foundE) * f(idx + 1, count, sum + a[idx], 1, foundE);
            if (foundE && count == 1 && idx == n)
            {
                res += f(idx + 1, count, sum + a[idx], 1, foundE);
            }
        }

        if (!foundE)
        {
            res += (count - foundB) * f(idx + 1, count, sum + a[idx], foundB, 1);
            if (foundB && count == 1 && idx == n)
            {
                res += f(idx + 1, count, sum + a[idx], foundB, 1);
            }
        }
    }

    if (count >= 2) 
    {
        int cntFree = count - foundB - foundE;
        res += 1LL * cntFree * (cntFree - 1) * f(idx + 1, count - 1, sum + 2 * a[idx], foundB, foundE);

        if (foundB)
        {
            res += 1LL * (count - 1 - foundE) * f(idx + 1, count - 1, sum + 2 * a[idx], foundB, foundE);
        }

        if (foundE)
        {
            res += 1LL * (count - 1 - foundB) * f(idx + 1, count - 1, sum + 2 * a[idx], foundB, foundE);
        }

        if (foundB && foundE && idx == n)
        {
            res += f(idx + 1, count - 1, sum + 2 * a[idx], foundB, foundE);
        }
    }

    return dp[idx][count][sum][foundB][foundE] = res % MOD;
}

void solve()
{
    if (n == 1)
    {
        std::cout << 1 << '\n';
        return;
    }

    std::sort(a + 1, a + 1 + n);
    std::cout << f(1, 0, MAXN * MAXL, 0, 0) << '\n';
}

void input()
{
    std::cin >> n >> l;
    for (int i = 1 ; i <= n ; ++i)
    {
        std::cin >> a[i];
    }
}

void fastIOI()
{
    std::ios_base:: sync_with_stdio(0);
    std::cout.tie(nullptr);
    std::cin.tie(nullptr);
}

int main()
{
    fastIOI();
    input();
    solve();

    return 0;
}

Subtask #1
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 604 KB Output is correct
5 Correct 1 ms 1628 KB Output is correct
6 Correct 1 ms 1564 KB Output is correct
7 Correct 1 ms 860 KB Output is correct
8 Correct 1 ms 604 KB Output is correct
9 Correct 1 ms 1372 KB Output is correct
10 Correct 1 ms 604 KB Output is correct

Subtask #2
15.0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 1112 KB Output is correct
2 Correct 3 ms 1884 KB Output is correct
3 Correct 1 ms 1372 KB Output is correct
4 Correct 3 ms 1884 KB Output is correct
5 Correct 5 ms 1880 KB Output is correct
6 Correct 2 ms 1628 KB Output is correct
7 Correct 2 ms 1628 KB Output is correct
8 Correct 1 ms 1372 KB Output is correct
9 Correct 1 ms 1484 KB Output is correct
10 Correct 3 ms 1884 KB Output is correct

Subtask #3
0 / 80.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 604 KB Output is correct
5 Correct 1 ms 1628 KB Output is correct
6 Correct 1 ms 1564 KB Output is correct
7 Correct 1 ms 860 KB Output is correct
8 Correct 1 ms 604 KB Output is correct
9 Correct 1 ms 1372 KB Output is correct
10 Correct 1 ms 604 KB Output is correct
11 Correct 1 ms 1112 KB Output is correct
12 Correct 3 ms 1884 KB Output is correct
13 Correct 1 ms 1372 KB Output is correct
14 Correct 3 ms 1884 KB Output is correct
15 Correct 5 ms 1880 KB Output is correct
16 Correct 2 ms 1628 KB Output is correct
17 Correct 2 ms 1628 KB Output is correct
18 Correct 1 ms 1372 KB Output is correct
19 Correct 1 ms 1484 KB Output is correct
20 Correct 3 ms 1884 KB Output is correct
21 Correct 1 ms 348 KB Output is correct
22 Incorrect 5 ms 2140 KB Output isn't correct
23 Halted 0 ms 0 KB -