Submission #61598404

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#include <bits/stdc++.h>
using namespace std;
const int N=500010;
int main()
{
  ios::sync_with_stdio(0);
  cin.tie(0);
  int a[N]={0},s[N]={0},b[N]={0};
  int n;
  cin>>n;
  for(int i=1;i<=n;i++)
    cin>>a[i];
  for(int i=1;i<=n;i++)
  {
    s[i]=s[i-1]+b[i];
    a[i]=a[i]+s[i];
    if(a[i]>0)
    {
      if(i+1<=n&&i+1+a[i]>n)
      {
        b[i+1]+=1;
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

#include <bits/stdc++.h>
using namespace std;
const int N=500010;
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	int a[N]={0},s[N]={0},b[N]={0};
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=1;i<=n;i++)
	{
		s[i]=s[i-1]+b[i];
		a[i]=a[i]+s[i];
		if(a[i]>0)
		{
			if(i+1<=n&&i+1+a[i]>n)
			{
				b[i+1]+=1;
			}
			else if(i+1<n&&i+1+a[i]<=n)
			{
				b[i+1]+=1;
				b[i+1+a[i]]-=1;
			}
		}
	}
	for(int i=1;i<=n;i++)
	{
		a[i]=a[i]-(n-i);
		if(a[i]<0)
			a[i]=0;
		cout<<a[i]<<" ";
	}
	cout<<endl;
	return 0;
}

Submission Info

Judge Result