Submission #365823

#	Time	Username	Problem	Language	Result	Execution time	Memory
365823		Ahmad_Hasan	Table Tennis (info1cup20_tabletennis)	C++17	35 / 100	3106 ms	1048580 KiB

tabletennis

#include <bits/stdc++.h>
#define int long long
/**
     ||||||||||       |||||     |||||    ||||||||||
    |||||||||||||     |||||     |||||  |||||
   ||||     ||||||    |||||     |||||  |||||
  |||||||||||||||||   |||||||||||||||    ||||||||||
 |||||||||||||||||||  |||||||||||||||           |||||
 |||||         |||||  |||||     |||||           |||||
 |||||         |||||  |||||     |||||    ||||||||||
AHMED;HASSAN;SAEED;
*/

using namespace std;
map<int,int>mp;
vector<int>v;
vector<vector<pair<int,pair<int,int> > > >dp;
int n,k;
int slv(int l,int r){
    ///cout<<l<<' '<<r<<'\n';
    if(dp[l][r].first!=-1)
        return dp[l][r].first;
    int ret=0;
    for(int i=l+1;i<r&&(i-l)<=k+5;i++){
        if(mp[v[r]-(v[i]-v[l])]>i+1){
            int ret2=1+slv(i,mp[v[r]-(v[i]-v[l])]-1);
            ///cout<<'#'<<i<<' '<<mp[v[r]-(v[i]-v[l])]-1<<'\n';
            if(ret2>ret){
                ret=ret2;
                dp[l][r].second={i,mp[v[r]-(v[i]-v[l])]-1};
            }

        }

    }

    return dp[l][r].first=ret;

}

int32_t main()
{
    /**ios_base::sync_with_stdio(0);
    cin.tie(0);      cout.tie(0);*/
    cin>>n>>k;
    n+=k;
    v=vector<int>(n);
    for(int i=0;i<n;i++){
        cin>>v[i];
        mp[v[i]]=i+1;
    }

    int ans=0;
    int l,r;
    dp=vector<vector<pair<int,pair<int,int>> > >(n+5,vector<pair<int,pair<int,int>> >(n+5,{-1,{-1,-1}}));
    for(int i=0;i<n;i++){
        for(int j=i;j<n;j++){
            int ret=slv(i,j);
            if(ret>ans){
                ans=ret;
                l=i,r=j;
            }


        }

    }
    vector<int>v2(n-k);
    int cnt=0;

    while(cnt<n-k){
        v2[cnt]=v[l],v2[cnt+1]=v[r];
        int nl=dp[l][r].second.first,nr=dp[l][r].second.second;
        l=nl,r=nr;
        cnt+=2;
    }
    sort(v2.begin(),v2.end());
    for(int i=0;i<n-k;i++)
        cout<<v2[i]<<' ';


    return 0;
}

Compilation message (stderr)

tabletennis.cpp: In function 'int32_t main()':
tabletennis.cpp:72:35: warning: 'r' may be used uninitialized in this function [-Wmaybe-uninitialized]
   72 |         v2[cnt]=v[l],v2[cnt+1]=v[r];
      |                                   ^
tabletennis.cpp:72:20: warning: 'l' may be used uninitialized in this function [-Wmaybe-uninitialized]
   72 |         v2[cnt]=v[l],v2[cnt+1]=v[r];
      |                    ^

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