Submission #855523

#	Time	Username	Problem	Language	Result	Execution time	Memory
855523		sofijavelkovska	Fancy Fence (CEOI20_fancyfence)	C++14	0 / 100	1 ms	760 KiB

fancyfence

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

const long long MAXN=100000, MOD=1000000007;

long long h[MAXN], w[MAXN];

bool compare(long long x, long long y)
{
    return h[x]<=h[y];
}

long long outer(long long i, long long sum)
{
    return (long long)h[i]*w[i]%MOD*sum%MOD;
}

long long inner(long long i)
{
    long long rectangles=0;
    rectangles=(rectangles+(long long)h[i]*w[i])%MOD;
    rectangles=(rectangles+h[i]*((long long)w[i]*(w[i]-1)/2%MOD)%MOD)%MOD;
    rectangles=(rectangles+w[i]*((long long)h[i]*(h[i]-1)/2%MOD)%MOD)%MOD;
    rectangles=(rectangles+((long long)h[i]*(h[i]-1)/2%MOD)*((long long)w[i]*(w[i]-1)/2%MOD)%MOD)%MOD;

    return rectangles;
}

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    long long n, x, left, right, i;
    long long leftsum, rightsum, total=0;
    set<long long> processed;
    cin >> n;
    long long index[i];
    long long prefixsum[n];
    for (i=0; i<n; i++)
        cin >> h[i];
    for (i=0; i<n; i++)
        cin >> w[i];
    for (i=0; i<n; i++)
        index[i]=i;
    sort(index, index+n, compare);
    prefixsum[0]=w[0];
    for (i=1; i<n; i++)
        prefixsum[i]=prefixsum[i-1]+w[i];
    processed.insert(-1);
    processed.insert(n);
    for (i=0; i<n; i++)
    {
        x=index[i];
        auto it=processed.lower_bound(x);
        right=*it;
        it--;
        left=*it;
        rightsum=(prefixsum[right-1]-prefixsum[x])%MOD;
        leftsum=(prefixsum[x]-w[x]-(prefixsum[left+1]-w[left+1]))%MOD;
        total=(total+outer(x, leftsum)+outer(x, rightsum)+inner(x))%MOD;
        processed.insert(x);
    }
    cout << total;

    return 0;
}

Compilation message (stderr)

fancyfence.cpp: In function 'int main()':
fancyfence.cpp:34:34: warning: 'i' may be used uninitialized in this function [-Wmaybe-uninitialized]
   34 |     long long n, x, left, right, i;
      |                                  ^

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