TAASEQ - editorial



Author: Tuấn Anh Trần Đặng
Tester: Kamil Dębowski
Editorialist: Tuấn Anh Trần Đặng




Dynamic Programming


Remove a single number in the sequence to make it an arithmetic progression.


The number at position i can be removed if the sub-sequence on its left and right are both arithmetic progression and when when we concat that two parts they still make an arithmetic progression.

The first condition can be handled by precalculate f[i] = whether the prefix of i characters of the sequence is arithmetic progression and similarly g[] for the suffixes. Put all the corner cases aside the whole conditions can be represented as f[i] and g[i] and a[i - 2] - a[i - 1] = a[i - 1] - a[i + 1] = a[i + 1] - a[i + 2] (1).


We will use dynamic programming (dp) to calculate f and g. Since they are similar let’s just discuss how to calculate f.
Initialize the dp with f[0] = f1 = f2 = true. We’ll have that

  • f[i] = (a[i - 1] - a[i] = a[i - 2] - a[i - 1] and f[i - 1]).

Before using the formula (1) some corner cases we need to consider is when N ≤ 3, i ≤ 2 and N - 2i.

Complexity is O(N).

Author’s/Tester’s Solutions:

Setter’s solution
Tester’s solution

1 Like

I actually had a different type of solution for this. Either we can remove index 1 and see if the remaining sequence is in AP or not. If it is we get one of the possible answers. Now we will definitely make sure index 1 lies in the answer. Now we remove index 2 and see if the sequence is in AP or not. If it is then we get one of the possible answers. Now we definitely include both index 1 and 2 in our solution. This fixes the common difference “d” for our case. Now simply iterate through the array and check for fault positions. If no faulty positions are found, this means the given sequence was already in AP. In such a case find the minimum value in the array which can be removed. If some faulty position is found break at the first one, as we can make exactly one removal. Again after removing these positions check if we get a valid AP or not and update the answer.

Answer is the minimum of all the above 4 cases.

Here is my solution https://www.codechef.com/viewsolution/11559787


I used the same approach while the contest was running as this editorial says . but got WA, I also took care of some boundary cases.
can anyone tell mistake in my code.?
my solution
( I am storing common diff. of the A.P. starting from 0 and ends at pos i in pre[i] else -1 )
thanks in advance :slight_smile:

The test cases for this problem are weak.

Here is my accpeted solution : https://www.codechef.com/viewsolution/11562158

but it fails on the following test case:



2 3 6 8 13 15 18

The answer for the above test case is -1 but my accepted solution gives 2 (link text)



Test cases are weak.

I have found accepted solutions of people with wrong answers.

Test Cases:



2 4 5 6 8

Given Ans: -1

Right Ans: 5



2 4 5 6

Given Ans: 5

Right Ans: 2

Some code are giving TLE.

I am checking through my compile list


2 3 6 8 13 15 18
3 5 7
7 5 3
1 7 12
12 7 1
2 6 7 8 11
2 7 12
2 5 7 11
3 6 7 11
2 4 6 15 8 10
2 4 5 6 8
2 4 5 6
1 1
4 5 6 8


Everyone check your answer against this.

I request to recheck all the submission.

1 Like

You can also store the count of difference between adjacent elements .

Suppose you have x,y,z and you remove y , the count of (z-y) and (y-x) would decrease by 1 and the count of (z-x ) would increase by 1 .

Now to check if the remaining sequence is an AP or not , all you need to check is that there exists a “difference” with count = n-2 .


can anybody explain me why we need to initialise the dp with f[0]=f[1]=f[2]=true ?

Please anybody explain why we need to intialise the dp with f[0]=f[1]=f[2]= true and why we took g[n+1]=g[n]=g[n-1]=true. Please make it clear

I Think i did this in a different way - https://ideone.com/Zj8LHm

can someone point out mistake in this solution.I tried all the test cases mentioned above.I have considered
three cases:
when a1=1St term and a[2] is second term of AP(deque x);
when a1=1st term and a[3] is second term of AP(deque y);
when a[2]=1st term and a[3] is second term of AP(deque z);

Its just a ad hoc for me… in this only 4 cases will be made…
Firstly array can be increasing order or decreasing order, so Make algo keeping in mind of smallest one.

  1. When N<4, then just take minimum of all them.
  2. Leave 1st element and check difference from arr[3] to arr[N-1] by taking d=arr[2]-arr[1], If loop iterates through whole element then just take only ans=arr[0](First element);
  3. Leave 2nd element and check difference from arr[3] to arr[N-1] by taking d=arr[2]-arr[0], If loop iterates through whole element then just take only ans=arr[1](2nd element);
  4. This case is boundary or will cover all others like ,d=arr[1]-arr[0]… Loop iterate from arr[2] to arr[N-1] and if condition fail one time then we just store the position of that element and iterate through loop… Here if loop iterates to end without breaking path then we just take minimum of 1st element and last element and print ans otherwise we print the element of that alteration…

To know more Just see my code…https://www.codechef.com/viewsolution/11570665

Can someone help me with my Java solution as well? I checked for corner cases, then stored the count of differences. If there was only one entry, then either first or last element is the answer. If more than three entries (if only only term is wrong then only 2 or 3 different kinds of differences can exist), then invalid solution. Else, I’ve found the difference which occurs maximum number of times and checked each pair of elements afterwards.


Can someone point out the mistake in my solution https://www.codechef.com/viewsolution/11575414
I have tried all test cases mentioned above, and it gives the expected answer for all.I divided the problem into three cases, if first element is to be removed, last element is to be removed or any other element except these two is to be removed to convert the sequence into AP.

Please add the question to practice section @admin @tuananh93

1 Like

I was getting TLE when I used cin - cout and with the statement “ios::sync_with_stdio(0);”. But then I simply changed these input output statements to scanf and printf: AC in 0.06 seconds.

Can someone rewrite this editorial with a better explanation ?

1 Like

can anyone help me with my python submission?i retried a few times and tested all the test cases i can think and find but it works correctly when i was using repl.it,however codechef just simply repeat ‘Wrong answer’,i cant get why
link text

You just have to store the difference of adjacent values and count them. If you remove y from x,y,z sequence then you have to check if count of [z-x] + 1 should be equal to N-2.

This is the simplest approach.

Here’s my approach, I created a difference array defined by d[i] = v[i+1] - v[i]. Then made a freq set which contains sorted list of pairs of(difference, its frequency) sorted by decreasing frequency.

Click here for submission

  1. If the first element has freq n-1 its already a AP so simply remove the min(first,last) element.
  2. If the first element has freq n-2, then there is 1 unequal element, now it can only be adjusted if that element is either at the front(remove the first element) or at the end(remove the last element) , else the answer is -1.
  3. If the first has freq n-3, then there’s 2 other differences. By property of d[] array, if I remove 1 element from v[], then the new d[] is is found by adding the 2 indices and merging it to one. So if the 2 differences does not add to the first element (with freq n-4) then it’s impossible so print -1. If it is then check if they are adjacent. If they are adjacent then remove the common element from v[] accordingly else can’t make it a AP and print -1

Note: For n=2,3 there is an edge case, it can be easily noted that removing any of the element makes it a valid AP. Thus remove the minimum of all element in this case.

-Thank You

Your code fails for this test case



2 4 6 8 10

Your code gives 10 but actual answer is 2 ( we have to output the smallest number)