oj.uz

답안 #874208

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
874208 2023-11-16T12:49:25 Z 12345678 앵무새 (IOI11_parrots) C++17
81 / 100
 2000 ms 138180 KB 
#include "encoder.h"
#include "encoderlib.h"
#include <bits/stdc++.h>
#pragma GCC optimize ("O3", "unroll-loops")
using namespace std;

const int nx=65, kx=256;

struct bignum
{
	vector<int> val;
	bignum(): val(nx, 0){}
	bignum(int x)
	{
		val.assign(nx, 0);
		for (int i=0; i<nx&&x>0; i++)
		{
			val[i]=x%kx;
			x/=kx;
		}
	}
	bignum &operator+=(const bignum &rhs)
	{
		int carry=0;
		for (int i=0; i<nx; i++)
		{
			val[i]+=rhs.val[i]+carry;
			carry=val[i]/kx;
			val[i]-=carry*kx;
		}
		return *this;
	}
	bignum &operator-=(const bignum &rhs)
	{
		int carry=0;
		for (int i=0; i<nx; i++)
		{
			val[i]-=rhs.val[i]+carry;
			carry=0;
			if (val[i]<0) {
				val[i]+=kx;
				carry++;
			}
		}
		return *this;
	}
	friend bignum operator+(const bignum &lhs, const bignum &rhs) {return bignum(lhs)+=rhs;}
	friend bignum operator-(const bignum &lhs, const bignum &rhs) {return bignum(lhs)+=rhs;}
	int cmp(const bignum &rhs) {
		for (int i=nx-1; i>=0; i--)
		{
			if (val[i]<rhs.val[i]) return -1;
			if (val[i]>rhs.val[i]) return 1;
		}
		return 0;
	}
	bool operator<(const bignum &rhs) {return cmp(rhs)==-1;}
	bool operator>(const bignum &rhs) {return cmp(rhs)==1;}
	bool operator<=(const bignum &rhs) {return cmp(rhs)!=1;}
	bool operator>=(const bignum &rhs) {return cmp(rhs)!=-1;}
	bool operator==(const bignum &rhs) {return cmp(rhs)==0;}
	bool operator!=(const bignum &rhs) {return cmp(rhs)!=0;}
};

vector<vector<bignum>> dp(580, vector<bignum> (270));
bool ok;

void encode(int N, int M[])
{
    bignum vl, tmp;
    if (!ok)
    {
        ok=1;
        for (int i=0; i<580; i++)
        {
            for (int j=0; j<=min(i, 269); j++)
            {
                if (j==0||j==i) dp[i][j]=1;
                else dp[i][j]=dp[i-1][j]+dp[i-1][j-1];
            }
        }
    }
    for (int i=0; i<N; i++) vl.val[i]=M[N-1-i];
    
    vector<int> s(kx+1);
    int left=320;
    for (int i=0; i<kx; i++)
    {
        tmp=0;
        for (int j=left; j>=0; j--)
        {
            if (dp[kx-i-1+left-j][kx-i-1]+tmp>vl)
            {
                vl-=tmp;
                s[i]=j;
                left-=j;
                break;
            }
            tmp+=dp[kx-i-1+left-j][kx-i-1];
        }
    }
    s[kx]=left;
    for (int i=0; i<kx; i++) for (int j=0; j<s[i+1]; j++) send(i);
}

#include "decoder.h"
#include "decoderlib.h"
#include <bits/stdc++.h>
#pragma GCC optimize ("O3", "unroll-loops")
using namespace std;

const int nx=65, kx=256;

struct bignum
{
	vector<int> val;
	bignum(): val(nx, 0){}
	bignum(int x)
	{
		val.assign(nx, 0);
		for (int i=0; i<nx&&x>0; i++)
		{
			val[i]=x%kx;
			x/=kx;
		}
	}
	bignum &operator+=(const bignum &rhs)
	{
		int carry=0;
		for (int i=0; i<nx; i++)
		{
			val[i]+=rhs.val[i]+carry;
			carry=val[i]/kx;
			val[i]-=carry*kx;
		}
		return *this;
	}
	bignum &operator-=(const bignum &rhs)
	{
		int carry=0;
		for (int i=0; i<nx; i++)
		{
			val[i]-=rhs.val[i]+carry;
			carry=0;
			if (val[i]<0) {
				val[i]+=kx;
				carry++;
			}
		}
		return *this;
	}
	friend bignum operator+(const bignum &lhs, const bignum &rhs) {return bignum(lhs)+=rhs;}
	friend bignum operator-(const bignum &lhs, const bignum &rhs) {return bignum(lhs)+=rhs;}
	int cmp(const bignum &rhs) {
		for (int i=nx-1; i>=0; i--)
		{
			if (val[i]<rhs.val[i]) return -1;
			if (val[i]>rhs.val[i]) return 1;
		}
		return 0;
	}
	bool operator<(const bignum &rhs) {return cmp(rhs)==-1;}
	bool operator>(const bignum &rhs) {return cmp(rhs)==1;}
	bool operator<=(const bignum &rhs) {return cmp(rhs)!=1;}
	bool operator>=(const bignum &rhs) {return cmp(rhs)!=-1;}
	bool operator==(const bignum &rhs) {return cmp(rhs)==0;}
	bool operator!=(const bignum &rhs) {return cmp(rhs)!=0;}
};

vector<vector<bignum>> dp(580, vector<bignum> (270));
bool ok2;

void decode(int N, int L, int X[])
{
    bignum res=0;
    if (!ok2)
    {
        ok2=1;
        for (int i=0; i<580; i++)
        {
            for (int j=0; j<=min(i, 269); j++)
            {
                if (j==0||j==i) dp[i][j]=1;
                else dp[i][j]=dp[i-1][j]+dp[i-1][j-1];
            }
        }
    }
    int sm=0;
    vector<int> s(kx+1, 0);
    for (int i=0; i<L; i++) s[X[i]+1]++, sm++;
    s[0]=320-sm;
    vector<vector<bignum>> dp(580, vector<bignum> (270));
    for (int i=0; i<580; i++)
    {
        for (int j=0; j<=min(i, 269); j++)
        {
            if (j==0||j==i) dp[i][j]=1;
            else dp[i][j]=dp[i-1][j]+dp[i-1][j-1];
        }
    }
    int left=320;
    for (int i=0; i<=kx; i++)
    {
        for (int j=left; j>s[i]; j--)
        {
            res+=dp[kx-i-1+left-j][kx-i-1];
        }
        left-=s[i];
    }
    //res+=s[kx];
    vector<int> ans(N, 0);
    for (int i=0; i<N; i++) ans[N-1-i]=res.val[i];
    for (int i=0; i<N; i++) output(ans[i]);
}

/*
10
1 1 2 1 2 3 1 7 10 9
*/

Subtask #1
17.0 / 17.0

# 결과 실행 시간 메모리 Grader output
1 Correct 478 ms 137212 KB Output is correct

Subtask #2
17.0 / 17.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1590 ms 137680 KB Output is correct
2 Correct 1624 ms 137440 KB Output is correct
3 Correct 1673 ms 137432 KB Output is correct
4 Correct 1650 ms 137804 KB Output is correct

Subtask #3
18.0 / 18.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1570 ms 137652 KB Output is correct
2 Correct 1611 ms 137612 KB Output is correct
3 Correct 1650 ms 137444 KB Output is correct
4 Correct 1656 ms 137568 KB Output is correct

Subtask #4
29.0 / 29.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1573 ms 137648 KB Output is correct
2 Correct 1648 ms 137720 KB Output is correct
3 Correct 1738 ms 137440 KB Output is correct
4 Correct 1787 ms 137768 KB Output is correct
5 Correct 1821 ms 137804 KB Output is correct
6 Correct 1819 ms 137584 KB Output is correct

Subtask #5
0.0 / 19.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1651 ms 137700 KB Output is correct - P = 1.750000
2 Correct 1811 ms 137188 KB Output is correct - P = 2.437500
3 Correct 1827 ms 137696 KB Output is correct - P = 2.484848
4 Execution timed out 2033 ms 137492 KB Time limit exceeded
5 Execution timed out 2064 ms 137988 KB Time limit exceeded
6 Execution timed out 2025 ms 138180 KB Time limit exceeded
7 Execution timed out 2076 ms 137968 KB Time limit exceeded