1604A - Era - CodeForces Solution

greedy *800

Please click on ads to support us..

Python Code: 

import sys

input = sys.stdin.readline


def solve():
    n = int(input())
    arr = list(map(int, input().split()))
    curr = 1
    ans = 0
    for i in range(n):
        if arr[i] <= curr:
            curr += 1
        else:
            ans += (arr[i] - curr)
            curr = arr[i] + 1
    return ans


for _ in range(int(input())):
    print(solve())

C++ Code: 

#include <bits/stdc++.h>
using namespace std;

typedef long long lol;

int main(){
    int t;
    cin>>t;
    while(t--){

        lol n;
        cin>>n;
        lol ar[n+5];
       
        for(int i=1;i<=n;i++)
            cin>>ar[i];

        lol ans = ar[1]-1;

        lol index = ans;

        lol j;
   

        for(int i=2;i<=n;i++){

            if((index+i)<ar[i]){

             j = ar[i] - (index+i);
                ans += j;
                index+=j;
            }
        }


        cout<<ans<<endl;

    }

}


Comments

Submit
0 Comments
More Questions

415. Add Strings
22. Generate Parentheses
13. Roman to Integer
2. Add Two Numbers
515. Find Largest Value in Each Tree Row
345. Reverse Vowels of a String
628. Maximum Product of Three Numbers
1526A - Mean Inequality
1526B - I Hate 1111
1881. Maximum Value after Insertion
237. Delete Node in a Linked List
27. Remove Element
39. Combination Sum
378. Kth Smallest Element in a Sorted Matrix
162. Find Peak Element
1529A - Eshag Loves Big Arrays
19. Remove Nth Node From End of List
925. Long Pressed Name
1051. Height Checker
695. Max Area of Island
402. Remove K Digits
97. Interleaving String
543. Diameter of Binary Tree
124. Binary Tree Maximum Path Sum
1465. Maximum Area of a Piece of Cake After Horizontal and Vertical Cuts
501A - Contest
160A- Twins
752. Open the Lock
1535A - Fair Playoff
1538F - Interesting Function