Cod sursa(job #3224014)

Utilizator xXoctavianXxStanescu Matei Octavian xXoctavianXx Data 14 aprilie 2024 12:40:44
Problema P-sir Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.76 kb
#include <bits/stdc++.h>
#define int long long

using namespace std;

ifstream fin("psir.in");
ofstream fout("psir.out");

const int mod = (1ll << 32) - 1;
const int nmax = 2005;
int v[nmax],d[nmax];
int n;

int aib[nmax];
void update(int p, int val)
{
    if(p == 0) return;
    while(p <= n + 1)
    {
        aib[p] += val;
        aib[p] &= mod;
        p += p & (-p);
    }
}
int query(int p)
{
    int ret = 0;
    while(p > 0)
    {
        ret += aib[p];
        ret &= mod;
        p -= p & (-p);
    }
    return ret;
}

inline void citire()
{
    fin>>n;
    int cnt=1;
    map<int,int> norm;
    for(int i=1; i<=n; i++)
    {
        fin>>v[i];
        norm[v[i]] = 1;
    }
    for(auto [i,j] : norm)
    {
        norm[i] = cnt++;
    }
    for(int i=1; i<=n; i++)
    {
        v[i] = norm[v[i]];
        //cout<<v[i]<<" ";
    }
}

signed main()
{
    citire();
    for(int i=2; i<=n; i++)
    {
        d[i] = i-1;
        d[i] &= mod;
        memset(aib,0,sizeof aib);
        //cout<<v[i]<<":\n";
        for(int j=1; j<i; j++)
        {
            //cout<<v[j]<<"-> ";
            if(v[j] > v[i])
            {
                //cout<<query(v[i] - 1)<<" ";
                d[i] += query(v[i] - 1);
            }
            if(v[j] < v[i])
            {
                //cout<<query(n + 1) - query(v[i])<<" ";
                d[i] += query(n + 1) - query(v[i]);
            }
            if(v[j] == v[i]) d[i] ++;
            //cout<<"\n";
            update(v[j],d[j] + 1);
            d[i] &= mod;
        }
    }

    int total = 0;
    for(int i=1; i<=n; i++)
    {
        //cout<<d[i]<<" ";
        total += d[i];
        total &= mod;
    }
    fout<<total;
    return 0;
}