Submission #594153

#	Time	Username	Problem	Language	Result	Execution time	Memory
594153		andrei_boaca	Nuclearia (CEOI15_nuclearia)	C++14	40 / 100	223 ms	262380 KiB

nuclearia

#include <bits/stdc++.h>
//#pragma GCC optimize("O3")
#define int long long
using namespace std;
typedef long long ll;
typedef long double ld;
vector<vector<ll>> v,S1,S2;
int n,m,k,q;
int nradds=0;
void add(int x1,int y1,int x2,int y2,ll val)
{
    x1=max(x1,1LL);
    y1=max(y1,1LL);
    x2=min(x2,n);
    y2=min(y2,m);
    v[x1][y1]+=val;
    v[x2+1][y1]-=val;
    v[x1][y2+1]-=val;
    v[x2+1][y2+1]+=val;
}
ll getsum(int x1,int y1,int x2,int y2)
{
    return v[x2][y2]-v[x1-1][y2]-v[x2][y1-1]+v[x1-1][y1-1];
}
ll put[2][2500005];
ll s[2500005];
signed main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>m>>k;
    if(m==1)
    {
        for(int z=1;z<=k;z++)
        {
            ll x,y,a,b;
            cin>>x>>y>>a>>b;
            //assert(y==1);
            ll nr=a/b;
            put[0][x]+=a;
            put[0][x+1]-=a;
            ll st=x-nr;
            ll dr=x-1;
            if(dr>0)
            {
                st=max(st,1LL);
                ll lg=dr-st+2;
                put[0][st]+=a-lg*b;
                put[0][dr+1]-=a-lg*b;
                put[0][dr+1]-=(lg-1)*b;
                put[1][st]+=b;
                put[1][dr+1]-=b;
            }
            st=x+1;
            dr=x+nr;
            if(st<=n)
            {
                dr=min(dr,n*1LL);
                ll lg=dr-st+2;
                put[0][st]+=a;
                put[0][dr+1]-=a;
                put[0][dr+1]-=(lg-1)*(-b);
                put[1][st]-=b;
                put[1][dr+1]+=b;
            }
        }
        ll suma=0;
        for(int i=1;i<=n;i++)
        {
            suma+=put[0][i];
            s[i]=suma;
        }
        ll si=0;
        suma=0;
        for(int i=1;i<=n;i++)
        {
            si+=put[1][i];
            suma+=si;
            s[i]+=suma;
        }
        for(int i=1;i<=n;i++)
            s[i]+=s[i-1];
        cin>>q;
        while(q--)
        {
            int x1,y1,x2,y2;
            cin>>x1>>y1>>x2>>y2;
            //assert(y1==1&&y2==1);
            ll val=s[x2]-s[x1-1];
            ll rez=round(ld(val)/ld(x2-x1+1));
            cout<<rez<<'\n';
        }
        return 0;
    }
    v.resize(n+5);
    S1.resize(n+5);
    S2.resize(n+5);
    for(int i=0;i<=n+4;i++)
    {
        v[i].resize(m+5);
        S1[i].resize(m+5);
        S2[i].resize(m+5);
    }
    for(int z=1;z<=k;z++)
    {
        int x,y,a,b;
        cin>>x>>y>>a>>b;
        int nr=a;
        int x1=x,y1=y,x2=x,y2=y;
        int cnt=a/b;
        /*if(cnt==0)
        {
            add(x,y,x,y,a);
            continue;
        }*/
        x1=x-cnt;
        x2=x+cnt;
        y1=y-cnt;
        y2=y+cnt;
        assert(x1>=1&&x1<=n);
        assert(y1>=1&&y1<=m);

        assert(x2>=1&&x2<=n);
        assert(y2>=1&&y2<=m);
        add(x1,y1,x2,y2,a);
        add(x1,y1,x2,y2,-(cnt+1)*b);
        /*while(x1<=x)
        {
            add(x1,y1,x2,y2,b);
            x1++;
            y1++;
            x2--;
            y2--;
        }*/
        S1[x1][y1]+=b;
        S1[x+1][y+1]-=b;
        S1[x+1][y+1]+=b;
        S1[x2+2][y2+2]-=b;

        if(y2+1<=m)
            S2[x1][y2+1]-=b;
        else
            S2[x1+1][y2]-=b;
        S2[x+1][y]+=b;
        S2[x+1][y]-=b;
        S2[x2+2][y1-1]+=b;
    }
    for(int i=1;i<=n;i++)
    {
        ll x=i,y=1;
        ll suma=0;
        while(x<=n&&y<=m)
        {
            suma+=S1[x][y];
            v[x][y]+=suma;
            x++;
            y++;
        }
    }
    for(int j=2;j<=m;j++)
    {
        ll x=1,y=j;
        ll suma=0;
        while(x<=n&&y<=m)
        {
            suma+=S1[x][y];
            v[x][y]+=suma;
            x++;
            y++;
        }
    }
    for(int i=1;i<=n;i++)
    {
        ll x=i,y=m;
        ll suma=0;
        while(x<=n&&y>0)
        {
            suma+=S2[x][y];
            v[x][y]+=suma;
            x++;
            y--;
        }
    }
    for(int j=1;j<m;j++)
    {
        ll x=1,y=j;
        ll suma=0;
        while(x<=n&&y>0)
        {
            suma+=S2[x][y];
            v[x][y]+=suma;
            x++;
            y--;
        }
    }
    /*for(int i=1;i<=n;i++,cout<<'\n')
        for(int j=1;j<=m;j++)
            cout<<v[i][j]<<' ';*/
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            v[i][j]+=v[i-1][j]+v[i][j-1]-v[i-1][j-1];
    /*for(int i=1;i<=n;i++,cout<<'\n')
        for(int j=1;j<=m;j++)
            cout<<v[i][j]<<' ';*/
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            v[i][j]=v[i][j]+v[i-1][j]+v[i][j-1]-v[i-1][j-1];
            //cout<<s[i][j]<<' ';
        }
    //assert(nradds<=k*max(n,m));
    cin>>q;
    while(q--)
    {
        int x1,y1,x2,y2;
        cin>>x1>>y1>>x2>>y2;
        ll suma=getsum(x1,y1,x2,y2);
        ll cells=(y2-y1+1)*(x2-x1+1);
        ll rez=round(ld(suma)/ld(cells));
        cout<<rez<<'\n';
    }
    return 0;
}

Compilation message (stderr)

nuclearia.cpp: In function 'int main()':
nuclearia.cpp:108:13: warning: unused variable 'nr' [-Wunused-variable]
  108 |         int nr=a;
      |             ^~

• Subtask 1: 3 / 3
• Subtask 2: 2 / 2
• Subtask 3: 0 / 3
• Subtask 4: 0 / 2
• Subtask 5: 6 / 6
• Subtask 6: 4 / 4
• Subtask 7: 0 / 6
• Subtask 8: 0 / 4
• Subtask 9: 15 / 15
• Subtask 10: 10 / 10
• Subtask 11: 0 / 15
• Subtask 12: 0 / 10
• Subtask 13: 0 / 12
• Subtask 14: 0 / 8