Submission #37197574

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
#define p_qu priority_queue
#define p_qu_less priority_queue<int, vector<int>, greater<int> >
#define C_in(a,n) for(int i=0;i<n;i++) cin>>a[i]
#define C_out(a,n) for(int i=0;i<n;i++) cout<<a[i]<<" "
#define SUM(a,n,sum) sum[0]=a[0];for(int i=1;i<n;i++) sum[i]=sum[i-1]+a[i];
#define SUM2(a,n,m,sum) sum[0][0]=a[0][0]; for(int i=0;i<n;i++) for(int j=0;j<m;j++) sum[i][j]=(i!=0?sum[i-1][j]:0)+(j!=0?sum[i][j-1]:0)+a[i][j]-(i!=0 && j!=0?sum[i-1][j-1]:0)
#define all(a) a.begin(),a.end()
#define l_b lower_bound
#define u_b upper_bound
#define pb push_back
#define max_3(a,b,c) max(max(a,b),c)
#define max_4(a,b,c,d) max(a,max_3(b,c,d))
#define min_3(a,b,c) min(min(a,b),c)
#define min_4(a,b,c,d) min(a,min_3(b,c,d))
#define zero(a) memset(a, 0, sizeof(a))
#define msit multiset<int>::iterator
#define setit set<int>::iterator
#define int long long
const int N=500005;
const int MAX=(1<<31)-1;
multiset<int> s,t;
int n,m,k;
int a[N];
int ans;
signed main()
{
	//freopen("input.txt","r",stdin);
	//freopen("output.txt","w",stdout);
	cin>>n>>m>>k;
	C_in(a,n);
	for(int i=0;i<m;i++)
	{
		t.insert(a[i]);
	}
	for(int i=0;i<k;i++)
	{
		s.insert(*t.begin());
		ans+=*t.begin();
		t.erase(t.begin());
	}
	cout<<ans<<" ";
	for(int i=1;i<=n-m;i++)
	{
		bool cc=false;
		if(a[i-1]>=*t.begin())
		{
			t.erase(t.find(a[i-1]));
		}
		else
		{
			s.erase(s.find(a[i-1]));
			ans-=a[i-1];
			cc=true;
		}
		int now=a[i+m-1];
		t.insert(now);
		if(cc)
		{
			s.insert(*t.begin());
			ans+=*t.begin();
			t.erase(t.begin());
		}
		else
		{
			if(*t.begin()<*s.rbegin())
			{
				ans-=*s.rbegin();
				ans+=*t.begin();
				msit j=s.end();
				j--;
				t.insert(*j);
				s.erase(j);
				msit i=t.begin();
				t.erase(i);
				s.insert(*i);
			}
		}
		cout<<ans<<" ";
	}
	return 0;
}

Submission Info

Compile Error

./Main.cpp:22:22: warning: integer overflow in expression of type ‘int’ results in ‘2147483647’ [-Woverflow]
   22 | const int MAX=(1<<31)-1;
      |               ~~~~~~~^~

Judge Result