Cod sursa(job #2910279)

Utilizator vlad2009Vlad Tutunaru vlad2009 Data 18 iunie 2022 22:54:45
Problema P-sir Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.96 kb
#include <fstream>
#include <iostream>
#include <algorithm>
#include <string.h>
#define int long long

using namespace std;

const int MAX_N = 2 * 1e3;
const long long MOD = (1LL << 32);
pair<int, int> a[MAX_N + 1];
long long dp[MAX_N + 1], index[MAX_N + 1], sum[MAX_N + 1];
unsigned int aib[MAX_N + 1][MAX_N + 1];
int n;

void update(long long x, long long y, long long val) {
    for (int i = y; i <= MAX_N; i += i & -i) {
       aib[x][i] = (aib[x][i] + val) % MOD;
    }
}

long long query(long long x, long long y) {
    long long answer = 0;
    for (int i = y; i >= 1; i -= i & -i) {
        answer = (answer + aib[x][i]) % MOD;
    }
    return answer;
}

signed main() {
    ifstream fin("psir.in");
    ofstream fout("psir.out");
    fin >> n;
    for (int i = 1; i <= n; i++) {
        fin >> a[i].first;
        a[i].second = i;
    }
    sort(a + 1, a + n + 1);
    int cnt = 2;
    index[a[1].second] = 2;
    for (int i = 2; i <= n; i++) {
        if (a[i].first != a[i - 1].first) {
            cnt++;
        }
        index[a[i].second] = cnt;
    }
    for (int i = 1; i <= n; i++) {
        memset(dp, 0, sizeof(dp));
        for (int j = 1; j < i; j++) {
            dp[index[j]]++;
        }
        for (int j = 2; j < i; j++) {
            if (index[j] > index[i]) {
                long long tmp = query(j, index[i] - 1);
                dp[index[j]] = (dp[index[j]] + tmp) % MOD;
            } else if (index[j] < index[i]) {
                long long tmp = (sum[j] - query(j, index[i]) + MOD) % MOD;
                dp[index[j]] = (dp[index[j]] + tmp) % MOD;
            }
        }
        for (int j = 1; j <= n + 1; j++) {
            if (dp[j] > 0) {
               update(i, j, dp[j]);
            }
            sum[i] = (sum[i] + dp[j]) % MOD;
        }
    }
    long long answer = 0;
    for (int i = 1; i <= n; i++) {
        answer = (answer + sum[i]) % MOD;
    }
    fout << answer;
    return 0;
}