Submission #780591

# Submission time Handle Problem Language Result Execution time Memory
780591 2023-07-12T10:35:33 Z vjudge1 Go (COCI18_go) C++17
50 / 100
157 ms 479632 KB
#include<bits/stdc++.h>
using namespace std;
#define lalala ios_base::sync_with_stdio(false);cin.tie(NULL);
//#define endl "\n"
#define ll long long
#define pb push_back
#define N 200005

int n,k,m;
vector<tuple<int,int,int>> pokemon;
ll int dp[3][2001][101][101];
ll int sagda(int ,int ,int ,int);
ll int solda(int nerden, int sn,int left,int right){

	if(sn>2000)return 0;
	int yedek=nerden;
	if(left<0||left>right)return 0;
	if(dp[nerden][sn][left][right]!=-1)return dp[nerden][sn][left][right];
	if(nerden==0)nerden=left-1;
	else if(nerden==1)nerden=right;
	else nerden=k;
	int e,d,f,x,y,z;
	if(nerden!=k)tie(f,d,e)=pokemon[nerden];
	else f=k;
	ll int ans=0,a=0,b=0;
	
	tie(x,y,z)=pokemon[left];	
	if(abs(x-f)+sn<z){
		a=y+solda(0,sn+abs(x-f),left-1,right);
		b=y+sagda(0,sn+abs(x-f),left,right+1);
		
	}
	else a=solda(0,sn+abs(x-f),left-1,right);
	//cout<<endl<<x<<" "<<y<<" "<<z<<" "<<f<<endl;
	//cout<<nerden<<" "<<yedek<<" a "<<sn<<" "<<left<<" "<<right<<" "<<a<<" "<<b<<endl;
	return dp[yedek][sn][left][right]=max(a,b);
}
ll int sagda(int nerden,int sn,int left,int right){
	if(sn>2000)return 0;
	int yedek=nerden;
	if(right>=m||left>right)return 0;
	if(dp[nerden][sn][left][right]!=-1&&nerden!=2)return dp[nerden][sn][left][right];
	if(nerden==0)nerden=left;
	else if(nerden==1)nerden=right-1;
	else nerden=k;
	int e,d,f,x,y,z;
	if(nerden!=k)tie(f,d,e)=pokemon[nerden];
	else f=k;
	ll int ans=0,a=0,b=0;
	tie(x,y,z)=pokemon[right];
	if(abs(x-f)+sn<z){
		a=y+sagda(1,sn+abs(x-f),left,right+1);
		b=y+solda(1,sn+abs(x-f),left-1,right);
		
	}
	else a=sagda(1,sn+abs(x-f),left,right+1);

	//cout<<endl<<x<<" "<<y<<" "<<z<<" "<<f<<endl;
	//cout<<nerden<<" "<<yedek<<" b "<<sn<<" "<<left<<" "<<right<<" "<<a<<" "<<b<<endl;
	return dp[yedek][sn][left][right]=max(a,b);
}
int main(){
	lalala;
	memset(dp,-1,sizeof(dp));
	int mm;cin>>n>>k>>mm;
	ll int cev=0;
	int le, rr;
	le=rr=-1;
	for(int i=0;i<mm;i++){
		ll int a,b,c;cin>>a>>b>>c;
		if(a==k){
			cev+=b;
			continue;
		}
		if(abs(k-a)<c){	
			pokemon.pb({a,b,c});
			if(a>k&&rr==-1)rr=m;
			else if(a<k)le=m;
			m++;
		}
	}
	if(rr==-1)rr=m+1;
	if(le==-1)le=-1;
	cout<<max(solda(2,0,le,rr-1),sagda(2,0,le,rr))+cev<<endl;



}

Compilation message

go.cpp: In function 'long long int solda(int, int, int, int)':
go.cpp:25:9: warning: unused variable 'ans' [-Wunused-variable]
   25 |  ll int ans=0,a=0,b=0;
      |         ^~~
go.cpp: In function 'long long int sagda(int, int, int, int)':
go.cpp:49:9: warning: unused variable 'ans' [-Wunused-variable]
   49 |  ll int ans=0,a=0,b=0;
      |         ^~~
# Verdict Execution time Memory Grader output
1 Correct 147 ms 479544 KB Output is correct
2 Correct 157 ms 479616 KB Output is correct
3 Incorrect 137 ms 479624 KB Output isn't correct
4 Correct 149 ms 479584 KB Output is correct
5 Correct 141 ms 479588 KB Output is correct
6 Incorrect 137 ms 479568 KB Output isn't correct
7 Incorrect 144 ms 479540 KB Output isn't correct
8 Correct 140 ms 479632 KB Output is correct
9 Incorrect 144 ms 479544 KB Output isn't correct
10 Incorrect 142 ms 479588 KB Output isn't correct