/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 324.0 KiB
#2 Accepted 1ms 404.0 KiB
#3 Accepted 2ms 320.0 KiB
#4 Accepted 410ms 3.605 MiB
#5 Accepted 1172ms 21.16 MiB
#6 Accepted 706ms 11.703 MiB
#7 Accepted 186ms 692.0 KiB
#8 Accepted 407ms 11.277 MiB
#9 Accepted 488ms 11.02 MiB
#10 Accepted 504ms 11.965 MiB
#11 Accepted 70ms 560.0 KiB
#12 Accepted 135ms 620.0 KiB
#13 Accepted 1ms 320.0 KiB

Code

#include<bits/stdc++.h>
using namespace std;
#define fast ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define dbg(a,b,c,d) cerr<<a<<"  "<<b<<"  "<<c<<"  "<<d<<endl;
#define kill(a) {cout<<a<<endl;continue;}
#define KILL(a) {cout<<a<<endl;return 0;}
#define debug cerr<<"Error Found"<<endl;
#define mem(a,b) memset(a,b,sizeof(a))
#define lcm(a, b) (a/__gcd(a,b))*b
#define w(t) cin>>t;while(t--)
#define pi  2 * acos(0.0)
#define endl "\n"
int t, cs = 0;
const int mxn = 1e6 + 3, mod = 1e9 + 7;
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
typedef tree<int,null_type,less_equal<int>,rb_tree_tag,tree_order_statistics_node_update>ordered_set;
int32_t main()
{

   w(t)
   {
       int n;
       cin >> n;
       ordered_set in1, in2, de1, de2;
       int ar[n];
       for(int i = 0; i < n; i++)cin >> ar[i], de1.insert(-ar[i]), de2.insert(ar[i]);
       int64_t ans[n], tmp;
       for(int i = 0; i < n; i++)
       {
           int a = in1.order_of_key(ar[i]);
           int b = de1.order_of_key(-ar[i]);
           tmp = 1LL * a * b;

           a = in2.order_of_key(-ar[i]);
           b = de2.order_of_key(ar[i]);
           tmp += 1LL * a * b;

           ans[i] = tmp;
           in1.insert(ar[i]), in2.insert(-ar[i]);
           de1.erase(--de1.lower_bound(-ar[i])), de2.erase(--de2.lower_bound(ar[i]));
       }
       int q;
       cin >> q;
       while(q--)
       {
           int idx;
           cin >> idx;
           cout << ans[--idx] << endl;
       }
   }
}

Information

Submit By
Type
Submission
Problem
P1079 Roy and Query (Easy Version)
Language
C++17 (G++ 13.2.0)
Submit At
2024-11-07 10:37:58
Judged At
2024-11-11 02:23:24
Judged By
Score
100
Total Time
1172ms
Peak Memory
21.16 MiB