oj.uz

Submission #405227

# Submission time Handle Problem Language Result Execution time Memory
405227 2021-05-16T02:06:09 Z varungoyalbits Fancy Fence (CEOI20_fancyfence) C++17
100 / 100
 37 ms 3472 KB 
/*
Problem Link:https://oj.uz/problem/view/CEOI20_fancyfence
Author: varungoyalbits
*/
#include <bits/stdc++.h>
using namespace std;
long long sumTillN(const long long N,const long long mod)
{
    return ((N*(N+1))/2)%mod;
}
long long subTask6()
{
    int n;
    cin>>n;
    vector<long long>h(n);
    vector<long long>w(n);
    long long mod=1e9+7;
    for(int i=0;i<n;i++)
    {
        cin>>h[i];
    }
    for(int i=0;i<n;i++)
    {
        cin>>w[i];
    }
    long long ans=0;
    for(int i=0;i<n;i++)
    {
        long long val=sumTillN(h[i],mod);
        val=(val*sumTillN(w[i],mod))%mod;
        ans=(ans+val)%mod;
        long long mn=h[i];
        for(int j=i-1;j>=0;j--)
        {
            mn=min(mn,h[j]);
            val=sumTillN(mn,mod);
            val=(val*((w[i]*w[j])%mod))%mod;
            ans=(ans+val)%mod;
        }
    }
    return ans;
}
long long sol()
{
    int n;
    cin>>n;
    vector<long long>h(n);
    vector<long long>w(n);
    vector<long long>w2(n);
    vector<int>left(n,-1);
    vector<int>right(n,n);
    long long mod=1e9+7;
    for(int i=0;i<n;i++)
    {
        cin>>h[i];
    }
    for(int i=0;i<n;i++)
    {
        cin>>w[i];
    }
    w2[0]=w[0];
    for(int i=1;i<n;i++)
    {
        w2[i]=(w2[i-1]+w[i])%mod;
    }
    for(int i=0;i<n;i++)
    {
        int ind=i-1;
        while(ind>=0&&h[ind]>=h[i])
        {
            ind=left[ind];
        }
        left[i]=ind;
    }
    for(int i=n-1;i>=0;i--)
    {
        int ind=i+1;
        while(ind<n&&h[ind]>h[i])
        {
            ind=right[ind];
        }
        right[i]=ind;
    }
    long long ans=0;
    for(int i=0;i<n;i++)
    {
        long long val=sumTillN(h[i],mod);
        val=(val*sumTillN(w[i],mod))%mod;
        ans=(ans+val)%mod;
    }
    //cout<<"ans: "<<ans<<endl;
    for(int i=0;i<n;i++)
    {
        int l=left[i];
        int r=right[i];
        //[l+1,i-1]
        //[i+1,r-1]
        //cout<<"l: "<<l+1<<" r: "<<r-1<<" i: "<<i<<endl;
        long long sumLeft=0;
        long long sumRight=0;
        if(l>=0&&i-1>=0)
        {
            sumLeft=w2[i-1]-w2[l];
        }
        else if(i-1>=0)
        {
            sumLeft=w2[i-1];
        }
        //cout<<"sumLeft: "<<sumLeft<<endl;

        if(r-1<=n-1)
        {
            sumRight=w2[r-1]-w2[i];
        }
        //cout<<"sumRight: "<<sumRight<<endl;
        sumLeft+=mod;
        sumLeft%=mod;
        sumRight+=mod;
        sumRight%=mod;
        long long totSum=(sumLeft+sumRight)%mod;
        long long val=sumTillN(h[i],mod);
        long long val2=(sumLeft*sumRight)%mod;
        long long val3=(w[i]*totSum)%mod;
        val2=(val2+val3)%mod;
        val=(val*val2)%mod;
        //cout<<"val: "<<val<<endl;
        ans=(ans+val)%mod;
        //cout<<"ans: "<<ans<<endl;
    }
    return ans;
}
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    //cout<<subTask6()<<"\n";
    cout<<sol()<<"\n";
    return 0;
}

Subtask #1
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 332 KB Output is correct

Subtask #2
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 204 KB Output is correct
3 Correct 1 ms 204 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 204 KB Output is correct

Subtask #3
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 332 KB Output is correct
3 Correct 13 ms 1892 KB Output is correct
4 Correct 26 ms 3472 KB Output is correct
5 Correct 26 ms 3464 KB Output is correct
6 Correct 30 ms 3404 KB Output is correct

Subtask #4
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 332 KB Output is correct
2 Correct 4 ms 588 KB Output is correct
3 Correct 16 ms 1892 KB Output is correct
4 Correct 30 ms 3404 KB Output is correct
5 Correct 32 ms 3448 KB Output is correct
6 Correct 1 ms 204 KB Output is correct

Subtask #5
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 332 KB Output is correct
3 Correct 4 ms 588 KB Output is correct
4 Correct 19 ms 1896 KB Output is correct
5 Correct 32 ms 3460 KB Output is correct
6 Correct 37 ms 3404 KB Output is correct
7 Correct 1 ms 332 KB Output is correct
8 Correct 4 ms 624 KB Output is correct
9 Correct 16 ms 1868 KB Output is correct
10 Correct 28 ms 3464 KB Output is correct
11 Correct 29 ms 3404 KB Output is correct
12 Correct 1 ms 204 KB Output is correct

Subtask #6
18.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 332 KB Output is correct
3 Correct 1 ms 204 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 204 KB Output is correct
6 Correct 1 ms 204 KB Output is correct
7 Correct 1 ms 204 KB Output is correct
8 Correct 1 ms 332 KB Output is correct
9 Correct 1 ms 332 KB Output is correct
10 Correct 1 ms 332 KB Output is correct
11 Correct 1 ms 204 KB Output is correct
12 Correct 1 ms 204 KB Output is correct
13 Correct 1 ms 332 KB Output is correct
14 Correct 1 ms 332 KB Output is correct
15 Correct 1 ms 332 KB Output is correct
16 Correct 1 ms 204 KB Output is correct

Subtask #7
27.0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 332 KB Output is correct
3 Correct 1 ms 204 KB Output is correct
4 Correct 1 ms 332 KB Output is correct
5 Correct 1 ms 204 KB Output is correct
6 Correct 1 ms 204 KB Output is correct
7 Correct 1 ms 204 KB Output is correct
8 Correct 1 ms 204 KB Output is correct
9 Correct 1 ms 204 KB Output is correct
10 Correct 1 ms 332 KB Output is correct
11 Correct 13 ms 1868 KB Output is correct
12 Correct 25 ms 3468 KB Output is correct
13 Correct 26 ms 3404 KB Output is correct
14 Correct 26 ms 3464 KB Output is correct
15 Correct 1 ms 332 KB Output is correct
16 Correct 4 ms 588 KB Output is correct
17 Correct 15 ms 1892 KB Output is correct
18 Correct 30 ms 3464 KB Output is correct
19 Correct 32 ms 3404 KB Output is correct
20 Correct 1 ms 332 KB Output is correct
21 Correct 4 ms 588 KB Output is correct
22 Correct 16 ms 1900 KB Output is correct
23 Correct 29 ms 3404 KB Output is correct
24 Correct 29 ms 3468 KB Output is correct
25 Correct 1 ms 204 KB Output is correct
26 Correct 1 ms 204 KB Output is correct
27 Correct 1 ms 332 KB Output is correct
28 Correct 1 ms 332 KB Output is correct
29 Correct 2 ms 332 KB Output is correct
30 Correct 4 ms 588 KB Output is correct
31 Correct 4 ms 588 KB Output is correct
32 Correct 16 ms 1892 KB Output is correct
33 Correct 16 ms 1868 KB Output is correct
34 Correct 30 ms 3404 KB Output is correct
35 Correct 30 ms 3404 KB Output is correct
36 Correct 32 ms 3468 KB Output is correct
37 Correct 31 ms 3404 KB Output is correct
38 Correct 1 ms 204 KB Output is correct
39 Correct 30 ms 3448 KB Output is correct
40 Correct 30 ms 3444 KB Output is correct
41 Correct 30 ms 3456 KB Output is correct
42 Correct 31 ms 3464 KB Output is correct