Submission #38016090

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;

int a[200100];
long long s[200100];
long long now;

int main()
{
    int n;

    cin>>n;
    for(int i=1; i<=n; i++)
        scanf("%d", &a[i]);

    for(int i=n; i>=1; i--) s[i]=s[i+1]+a[i];
    for(int i=1; i<=n; i++) now+=a[i]*i;

    for(int i=1; i<=n; i++)
    if(s[i])
    {
        long long t=0,tnow=now;
        if(tnow>0&&s[i]>0)
        {
            if(!s[1]) continue;
            long long tmp=abs(s[1]);
            t=tnow/tmp; tnow%=tmp;
            if(tnow>0) ++t, tnow-=tmp;
            if(s[1]>0) t=-t;
        }
        if(tnow<0&&s[i]<0)
        {
            if(!s[1]) continue;
            long long tmp=abs(s[1]);
            tnow=-tnow;
            t=tnow/tmp; tnow%=tmp;
            if(tnow>0) ++t, tnow-=tmp;
            if(s[1]<0) t=-t;
            tnow=-tnow;
        }
        if(tnow%s[i]&&!s[1]) continue;
        long long ft=abs(tnow/s[i]); tnow%=s[i];
        if(tnow&&tnow%s[1]==0) t-=tnow/s[1];
        else if(tnow&&(s[i]+tnow)%s[1]==0)
        {
            ++ft;
            t-=(s[i]+tnow)/s[1];
        }
        else if(tnow) continue;
        puts("Yes");
        for(long long j=1; j<i; j++)
            printf("%lld ", j+t);
        for(long long j=i; j<=n; j++)
            printf("%lld ", j+t+ft);
        return 0;
    }
    puts("No");
    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:14:14: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   14 |         scanf("%d", &a[i]);
      |         ~~~~~^~~~~~~~~~~~~

Judge Result