Submission #791359

#	Time	Username	Problem	Language	Result	Execution time	Memory
791359		Username4132	Izbori (COCI22_izbori)	C++14	0 / 110	16 ms	5004 KiB

Main

#include<iostream>
#include<map>
#include<vector>
using namespace std;
using pii = pair<int, int>;
using ll = long long;
#define forn(i, n) for(int i=0; i<(int)n; ++i)
#define PB push_back
#define F first
#define S second

const int MAXN = 200010;
int n, arr[MAXN];
ll ans;
pii calc[3*MAXN];
map<int, vector<int>> mp;

int main(){
    scanf("%d", &n);
    forn(i, n) scanf("%d", arr+i), mp[arr[i]].PB(i);
    forn(i, 3*n+10) calc[i]={-1, -1};
    for(auto pr:mp){
        int sz = pr.S.size();
        vector<int> toClear;
        forn(i, sz){
            int pa = pr.S[i], bnd = (i==sz-1)? n : pr.S[i+1];
            bool yes=true;
            while(pa<bnd){
                int val = 2*i-pa+n;
                if(calc[val].F==-1){
                    if(!yes) break;
                    if(val>=n-pr.S[0]) calc[val]={min(2*i+1, pa+1), pa};
                    else calc[val]={1, pa};
                    toClear.PB(val);
                }
                else{
                    if(yes) calc[val].F+=pa-calc[val].S-1;
                    calc[val].S=pa;
                }
                ans+=calc[val].F;
                yes=false;
                ++pa;
            }
        }
        for(int el:toClear) calc[el]={-1, -1};
    }
    printf("%lld\n", ans);
}

Compilation message (stderr)

Main.cpp: In function 'int main()':
Main.cpp:19:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   19 |     scanf("%d", &n);
      |     ~~~~~^~~~~~~~~~
Main.cpp:20:21: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   20 |     forn(i, n) scanf("%d", arr+i), mp[arr[i]].PB(i);
      |                ~~~~~^~~~~~~~~~~~~

• Subtask 1: 0 / 10
• Subtask 2: 0 / 15
• Subtask 3: 0 / 15
• Subtask 4: 0 / 70