Submission #274857

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
274857	2020-08-19T17:31:07 Z	arnold518	Teams (CEOI11_tea)	C++14	80 / 100	577 ms	151544 KB

#include <bits/stdc++.h>
using namespace std;
 
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
 
const int MAXN = 1e6;
const int INF = 987654321;

struct dequeue
{
	int st;
	vector<int> V;
	dequeue() : st(0) {} 
	bool empty() { return st==V.size(); }
	void push_back(int x) { V.push_back(x); }
	void push_front(int x) { V[--st]=x; }
	void pop_back() { V.pop_back(); }
	void pop_front() { st++; }
	int front() { return V[st]; }
	int back() { return V.back(); }
};

int N;
pii B[MAXN+10];
int dp1[MAXN+10], dp2[MAXN+10];
int pmax[MAXN+10];
dequeue DQ[MAXN+10];
int memo[MAXN+10]; 
vector<int> V[MAXN+10];

int main()
{
	scanf("%d", &N);
	for(int i=1; i<=N; i++) scanf("%d", &B[i].first), B[i].second=i;
	//for(int i=1; i<=N; i++) A[i]=rand()%(N/2)+1;
	sort(B+1, B+N+1);
 
	for(int i=1; i<=N; i++)
	{
		dp1[i]=-INF;
		if(i-B[i].first>=0)
		{
			dp1[i]=pmax[i-B[i].first]+1;
			V[i-B[i].first].push_back(i);
		}
		pmax[i]=max(pmax[i-1], dp1[i]);
	}

	for(int i=0; i<=N; i++)
	{
		if(dp1[i]!=-INF)
		{
			while(!DQ[dp1[i]].empty() && dp2[DQ[dp1[i]].back()]>=dp2[i]) DQ[dp1[i]].pop_back();
			DQ[dp1[i]].push_back(i);
		}
		for(auto it : V[i])
		{
			int p=pmax[i];
			int val=-1;
			while(!DQ[p].empty() && dp2[DQ[p].front()]<=it-DQ[p].front())
			{
				val=DQ[p].front();
				DQ[p].pop_front();
			}
			if(val!=-1 && (DQ[p].empty() || max(it-val, dp2[val])<dp2[DQ[p].front()])) DQ[p].push_front(val);
			dp2[it]=max(it-DQ[p].front(), dp2[DQ[p].front()]);
			memo[it]=DQ[p].front();
		}
	}

	printf("%d\n", dp1[N]);
	int now=N; int cnt=0;
	while(now)
	{
		printf("%d ", now-memo[now]); cnt++;
		for(int i=now; i>memo[now]; i--) printf("%d ", B[i].second);
		printf("\n"); now=memo[now];
	}
	assert(cnt==dp1[N]);
}

Compilation message

tea.cpp: In member function 'bool dequeue::empty()':
tea.cpp:16:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   16 |  bool empty() { return st==V.size(); }
      |                        ~~^~~~~~~~~~
tea.cpp: In function 'int main()':
tea.cpp:35:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   35 |  scanf("%d", &N);
      |  ~~~~~^~~~~~~~~~
tea.cpp:36:31: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   36 |  for(int i=1; i<=N; i++) scanf("%d", &B[i].first), B[i].second=i;
      |                          ~~~~~^~~~~~~~~~~~~~~~~~~

Subtask #1
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	34 ms	55168 KB	Output is correct
2	Correct	38 ms	55124 KB	Output is correct
3	Correct	35 ms	55168 KB	Output is correct
4	Correct	34 ms	55160 KB	Output is correct
5	Correct	36 ms	55168 KB	Output is correct

Subtask #2
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	33 ms	55168 KB	Output is correct
2	Correct	33 ms	55168 KB	Output is correct
3	Correct	35 ms	55168 KB	Output is correct

Subtask #3
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	33 ms	55160 KB	Output is correct
2	Correct	34 ms	55296 KB	Output is correct
3	Correct	34 ms	55168 KB	Output is correct

Subtask #4
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	35 ms	55416 KB	Output is correct
2	Correct	36 ms	55460 KB	Output is correct
3	Correct	44 ms	55416 KB	Output is correct

Subtask #5
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	37 ms	55424 KB	Output is correct
2	Correct	37 ms	55424 KB	Output is correct

Subtask #6
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	67 ms	59512 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #7
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	73 ms	60412 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #8
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	382 ms	101368 KB	Output is correct
2	Correct	362 ms	102136 KB	Output is correct
3	Correct	373 ms	99572 KB	Output is correct

Subtask #9
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	490 ms	114532 KB	Output is correct
2	Correct	577 ms	151544 KB	Output is correct
3	Correct	472 ms	117240 KB	Output is correct
4	Correct	462 ms	108568 KB	Output is correct

Subtask #10
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	483 ms	100200 KB	Output is correct
2	Correct	355 ms	77688 KB	Output is correct
3	Correct	479 ms	116856 KB	Output is correct
4	Correct	523 ms	115160 KB	Output is correct