Submission #944921

#	Time	Username	Problem	Language	Result	Execution time	Memory
944921		sleepntsheep	Fancy Fence (CEOI20_fancyfence)	C11	25 / 100	17 ms	6488 KiB

fancyfence

#include<stdio.h>

#define int long long
#define N 100002
#define M 1000000007ll
int n,h[N],w[N],p[N],l[N],r[N];
long long pw[N],z;

long long c(long long ww,long long hh)
{
    ww%=M;
    hh%=M;
    return (((ww+1ll)*ww/2)%M*((hh+1ll)*hh/2)%M)%M;
}

void dfs(int u,int ll,int rr,int yy)
{
    int y=h[u];
    if(!u||ll>rr)return;
    long long ww=(pw[rr]-pw[ll-1]);
    z = (z + (c(ww,y) + M - c(ww,yy)) % M) % M;
    //printf(" AT %lld : [%lld %lld], %lld %lld\n",u,pw[ll-1]+1,pw[rr],y,z);
    dfs(l[u],ll,u-1,y);
    dfs(r[u],u+1,rr,y);
}

int stk[N], top;

signed main()
{
    scanf("%lld",&n);
    for(int i=1;i<=n;++i)scanf("%lld",h+i);
    for(int i=1;i<=n;++i)scanf("%lld",w+i),pw[i]=pw[i-1]+w[i];

    //for(int i=1;i<=n;++i)printf(" %d %d %d\n",i,l[i],r[i]);
    for(int i=1;i<=n;++i)
    {
        int k=top;
        while(k&&h[stk[k]]>h[i])--k;
        if(k) r[stk[k]]=i;
        if(k<top)l[i]=stk[k+1];
        stk[++k]=i;
        top=k;
    }

    dfs(stk[1],1,n,0);
    printf("%lld",z);
}

Compilation message (stderr)

fancyfence.c: In function 'main':
fancyfence.c:31:5: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   31 |     scanf("%lld",&n);
      |     ^~~~~~~~~~~~~~~~
fancyfence.c:32:26: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   32 |     for(int i=1;i<=n;++i)scanf("%lld",h+i);
      |                          ^~~~~~~~~~~~~~~~~
fancyfence.c:33:26: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   33 |     for(int i=1;i<=n;++i)scanf("%lld",w+i),pw[i]=pw[i-1]+w[i];
      |                          ^~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 0
• Subtask 2: 12 / 12
• Subtask 3: 13 / 13
• Subtask 4: 0 / 15
• Subtask 5: 0 / 15
• Subtask 6: 0 / 18
• Subtask 7: 0 / 27