Перевести з Pascal в Python очень нужно

#include <iostream>

#include <vector>

#include <set>

#define ll long long

using namespace std;

signed main() {

   ll n;

   cin >> n;

   vector<pair<ll,ll>> a(n);

   vector<ll> pref(n,0),d(n,0),ans(n,0);

   set<ll> s;

   for(ll i = 0; i < n; i++){

       cin >> a[i].first;

       a[i].second = i;

       s.insert(a[i].first);

       if(i == 0)

           pref[i] = a[i].first;

       else

           pref[i] = pref[i-1] + a[i].first;

       d[i] = s.size();

   }

   if(d[n-1] > 1 || n == 1)

       ans[a[n-1].second] = 1;

   for(ll i = n - 2; i >= 0; i--){

       if(pref[i] > a[i + 1].first && ans[a[i+1].second] == 1 && d[i] > 1)

           ans[a[i].second] = 1;

   }

   for(ll i = 0; i < n; i++)

       cout << ans[i] << " ";

}

P.S. Откуда задача, уж больно часто я ее встречаю на подобных сайтах

При  решении  целого  ряда  задач  Леонард  Эйлер  использовал  идею  изображения  множеств  с кругов  и  они  получили  название  «круги  Эйлера». 

С этих  кругов  Эйлер  изобразил  и  множество  всех  действительных  чисел: 

·     N  —  множество  натуральных  чисел, 

·     Z  —  множество  целых  чисел, 

·     Q  —  множество  рациональных  чисел, 

·     R  —  множество  всех  действительных  чисел. 

 
В  нашем  классе  24  ученика.  Все  они  хорошо  провели  зимние  каникулы.10  человек  катались  на  лыжах,  16  ездили  на  каток,  а  12  —  лепили  снеговиков.  Сколько  учеников  смогли  покататься  и  на  лыжах,  и  на  коньках,  и  слепить  снеговика?

А  —  множество  ребят,  катающихся  на  лыжах

В  —  множество  ребят,  катающихся  на  коньках

С  —  множество  ребят,  лепивших  снеговиков

Решение

Пусть  х  —  число  ребят, 

которые  успели  за  эти  каникулы  всё!

(12  -  х)  +  (16  -  х)  +  (10  -  х)  +  х  =  24

ответ:  7  ребят