Submission #1242829

#	Time	Username	Problem	Language	Result	Execution time	Memory
1242829		MasterDebater	UFO (IZhO14_ufo)	C++20	35 / 100	2096 ms	23964 KiB

ufo

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll n,m,r,k,p,ans;
vector<vector<ll> >a,dp;
ll calc(int a1,int b1,int a2,int b2){
	if(a1==0 and b1==0)return dp[a2][b2];
	if(a1==0)return dp[a2][b2]-dp[a2][b1-1];
	if(b1==0)return dp[a2][b2]-dp[a1-1][b2];
	return dp[a2][b2]-dp[a1-1][b2]-dp[a2][b1-1]+dp[a1-1][b1-1];
}
int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	cin>>n>>m>>r>>k>>p;
	for(int i=0;i<n;i++){
		vector<ll>v;
		for(int j=0;j<m;j++){
			int ai;
			cin>>ai;
			v.push_back(ai);
		}
		dp.push_back(v);
		a.push_back(v);
	}
	for(int i=0;i<k;i++){
		char c;
		int x,h;
		cin>>c>>x>>h,x--;
		if(c=='N'){
			for(int j=0,cnt=r;j<n and cnt>0;j++)if(a[j][x]>=h)a[j][x]--,cnt--;
		}
		else if(c=='S'){
			for(int j=n-1,cnt=r;j>=0 and cnt>0;j--)if(a[j][x]>=h)a[j][x]--,cnt--;
		}
		else if(c=='W'){
			for(int j=0,cnt=r;j<m and cnt>0;j++)if(a[x][j]>=h)a[x][j]--,cnt--;
		}
		else if(c=='E'){
			for(int j=m-1,cnt=r;j>=0 and cnt>0;j--)if(a[x][j]>=h)a[x][j]--,cnt--;
		}
	}
	/*for(int i=0;i<n;i++){
		for(int j=0;j<m;j++)cout<<a[i][j]<<' ';
		cout<<'\n';
	}*/
	for(int i=0;i<n;i++)for(int j=0;j<m;j++)dp[i][j]=0;
	dp[0][0]=a[0][0];
	for(int i=1;i<n;i++)dp[i][0]=dp[i-1][0]+a[i][0];
	for(int j=1;j<m;j++)dp[0][j]=dp[0][j-1]+a[0][j];
	for(int i=1;i<n;i++)for(int j=1;j<m;j++)dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+a[i][j];
	/*for(int i=0;i<n;i++){
		for(int j=0;j<m;j++)cout<<dp[i][j]<<' ';
		cout<<'\n';
	}*/
	for(int i=0;i<=n-p;i++)for(int j=0;j<=m-p;j++){
		ans=max(ans,calc(i,j,i+p-1,j+p-1));
		//if(calc(i,j,i+p-1,j+p-1)>6)cout<<i+1<<' '<<j+1<<'\n';
	}
	cout<<ans;
	return 0;
}

• Subtask 1: 35 / 100