Submission #942227

#	Time	Username	Problem	Language	Result	Execution time	Memory
942227		Litusiano	Pilot (NOI19_pilot)	C++17	45 / 100	75 ms	31056 KiB

pilot

#include<bits/stdc++.h>
using namespace std;
// #define int long long
#define endl "\n"
const int MAXN = 1e6+5;

long long calc(pair<int,int> p){
	int len = p.second - p.first +1;
	return 1LL*(len*(len+1))/2;
}

signed main(){
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	int n,m; cin>>n>>m;
	vector<int> v(n); for(int& i : v) cin>>i;
	vector<int> b(m); for(int& i : b) cin>>i;
	set<pair<int,int>> segs; segs.insert({0,n-1}); segs.insert({n,n});
	vector<vector<int>> ST(MAXN);
	for(int i = 0; i<m; i++) ST[b[i]].push_back(i); // indexed one more
	for(int i = 0; i<n; i++) ST[v[i]].push_back(-i-1);
	long long ans = 1LL*n*(n+1)/2;
	vector<long long> q(m);
	// cerr<<MAXN<<endl;
	for(int j = MAXN-1; j>=0; j--){
		for(auto i : ST[j]){
			if(i < 0){
				// cur seg
				if(segs.size() == 0) continue;
				int x = -i -1;//i+1
				pair<int,int> p = {x, n};
				auto it = segs.upper_bound(p);
				if(it != segs.begin()) it = prev(it);
				pair<int,int> p1 = *it;
				// divide p1 in two
				int len = p1.second - p1.first +1;
				ans -= 1LL*(len*(len+1))/2;
				pair<int,int> seg1; pair<int,int> seg2;
				seg1 = {p1.first, x-1}; seg2 = {x+1, p1.second};
				// cerr<<"X: "<<x<<" J: "<<j<<" P1 "<<p1.first<<" "<<p1.second<<" SEG1 "<<seg1.first<<" "<<seg1.second<<" SEG2 "<<seg2.first<<" "<<seg2.second<<endl;
				if(seg1.second >= seg1.first) segs.insert(seg1), ans+= calc(seg1);
				if(seg2.second >= seg2.first) segs.insert(seg2), ans+= calc(seg2);
				segs.erase(p1);
				// cerr<<"SEGS: ";
				// for(auto z : segs) cout<<z.first<<" "<<z.second<<"  "; cout<<endl;
			}
			else{
				q[i] = ans;
			}
		}
	}
	for(long long i : q) cout<<i<<endl;
}

• Subtask 1: 3 / 3
• Subtask 2: 10 / 10
• Subtask 3: 12 / 12
• Subtask 4: 15 / 15
• Subtask 5: 5 / 5
• Subtask 6: 0 / 9
• Subtask 7: 0 / 14
• Subtask 8: 0 / 10
• Subtask 9: 0 / 11
• Subtask 10: 0 / 11