oj.uz

Submission #877958

# Submission time Handle Problem Language Result Execution time Memory
877958 2023-11-23T22:54:21 Z Ice_man Skyscraper (JOI16_skyscraper) C++14
5 / 100
 235 ms 524288 KB 
#include <iostream>
#include <chrono>
#include <algorithm>
#include <cstring>

#define maxn 105
#define maxdiff 1005
#define maxlog 20
#define INF 1000000010
#define LINF 1000000000000000005
#define mod 1000000007
#define endl '\n'
#define pb(x) push_back(x)
#define X first
#define Y second
#define control cout<<"passed"<<endl;

#pragma GCC optimize("O3" , "Ofast" , "fast-math" , "unroll-loops")
#pragma GCC target(avx2)

using namespace std;

std::chrono::high_resolution_clock::time_point startT, currT;
constexpr double TIME_MULT = 1;

double timePassed()
{
    using namespace std::chrono;
    currT = high_resolution_clock::now();
    double time = duration_cast<duration<double>>(currT - startT).count();
    return time * TIME_MULT;
}

int dp[(1 << 15)][15][500];
int val[maxn];
///maxdiff!!!

int n , max_sum;

void read()
{
    cin >> n >> max_sum;
    for(int i = 1; i <= n; i++) cin >> val[i];

    sort(val + 1 , val + 1 + n);
    val[0] = val[1];
}

int ans = 0;

void subtask1()
{
    sort(val + 1 , val + 1 + n);
    int cur_sum = 0;
    do
    {
        cur_sum = 0;

        for(int i = 2; i <= n; i++) cur_sum += abs(val[i - 1] - val[i]);

        if(cur_sum <= max_sum) ans++;

    }while(next_permutation(val + 1 , val + 1 + n));

    cout << ans << endl;

}

void subtask2()
{

    dp[0][0][0] = 1;

    for(int mask = 0; mask < (1 << n) - 1; mask++)
    {
        for(int i = 0; i <= n; i++)
        {
            for(int j = 0; j <= max_sum; j++)
            {
                if(dp[mask][i][j])
                {

                    for(int k = 1; k <= n; k++)
                    {
                        if(!(mask & (1 << (k - 1))))
                        {

                            int new_mask = (mask | (1 << (k - 1)));

                            int pom;
                            if(mask == 0) pom = 0;
                            else pom = j + abs(val[i] - val[k]);

                            dp[new_mask][k][pom] = (dp[new_mask][k][pom] + dp[mask][i][j]) % mod;


                        }
                    }

                }
            }
        }
    }


    for(int i = 1; i <= n; i++)
        for(int j = 0; j <= max_sum; j++)
        ans = (ans + dp[(1 << n) - 1][i][j]) % mod;

    cout << ans << endl;

}

int dp2[maxn][maxn][maxdiff][2][2];

int subtask3(int sum , int l , int r , int i , int j)
{

    int new_sum = sum + (j * 2 + l + r) * (val[i] - val[i - 1]);
    if(new_sum > max_sum) return 0;

    if(i == n) return j == 0? 1 : 0;

    if(dp2[i][j][sum][l][r] != -1) return dp2[i][j][sum][l][r];

    int cur_ans = 0;

    cur_ans += subtask3(new_sum , l , 1 , i + 1, j);         ///dqsno
    cur_ans += subtask3(new_sum , l , 1 , i + 1 , j - 1);

    cur_ans += subtask3(new_sum , 1 , r , i + 1 , j);        ///lqvo
    cur_ans += subtask3(new_sum , 1 , r , i + 1 , j - 1);

    cur_ans += subtask3(new_sum , l , r , i + 1 , j) * 2 * j; /// dobavi

    cur_ans += subtask3(new_sum , l , r , i + 1 , j + 1); ///novo

    cur_ans += subtask3(new_sum , l , r , i + 1 , j - 1) * (j - 1) * j; ///mergeni

    cur_ans %= mod;
    return (cur_ans % mod);
}




int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    startT = std::chrono::high_resolution_clock::now();

    ///control

    read();

    memset(dp2 , -1 , sizeof(dp2));

    if(n <= 8) subtask1();
    else if(n <= 14) subtask2();
    else cout << subtask3(0 , 0 , 0 , 1 , 0) << endl;

    return 0;
}

Compilation message

skyscraper.cpp:19:20: warning: '#pragma GCC option' is not a string [-Wpragmas]
   19 | #pragma GCC target(avx2)
      |                    ^~~~

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 23 ms 174936 KB Output is correct
2 Correct 19 ms 174940 KB Output is correct
3 Correct 20 ms 174936 KB Output is correct
4 Correct 19 ms 174940 KB Output is correct
5 Correct 21 ms 174940 KB Output is correct
6 Correct 21 ms 174984 KB Output is correct
7 Correct 20 ms 174884 KB Output is correct
8 Correct 21 ms 174940 KB Output is correct
9 Correct 21 ms 175288 KB Output is correct
10 Correct 19 ms 174936 KB Output is correct

Subtask #2
0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 62 ms 249948 KB Output is correct
2 Runtime error 235 ms 524288 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #3
0 / 80.0

# Verdict Execution time Memory Grader output
1 Correct 23 ms 174936 KB Output is correct
2 Correct 19 ms 174940 KB Output is correct
3 Correct 20 ms 174936 KB Output is correct
4 Correct 19 ms 174940 KB Output is correct
5 Correct 21 ms 174940 KB Output is correct
6 Correct 21 ms 174984 KB Output is correct
7 Correct 20 ms 174884 KB Output is correct
8 Correct 21 ms 174940 KB Output is correct
9 Correct 21 ms 175288 KB Output is correct
10 Correct 19 ms 174936 KB Output is correct
11 Correct 62 ms 249948 KB Output is correct
12 Runtime error 235 ms 524288 KB Execution killed with signal 9
13 Halted 0 ms 0 KB -