Dynamic-Programming/K-orderedLCS(Rec).cpp at master · robinrst/Dynamic-Programming · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
#include        "algorithm"
#include        "iostream"
#include        "numeric"
#include        "iomanip"
#include        "cstring"
#include        "math.h"
#include        "bitset"
#include        "string"
#include        "vector"
#include        "ctime"
#include        "queue"
#include        "stack"
#include        "map"
#include        "set"

#include        "ext/pb_ds/assoc_container.hpp" // Common file
#include        "ext/pb_ds/tree_policy.hpp" // Including tree_order_statistics_node_update
#include        "ext/pb_ds/detail/standard_policies.hpp"

using namespace std;
using namespace __gnu_pbds;


#define          f first
#define          lgn 25
#define          endl '\n'
#define          sc second
#define          pb push_back
#define          N (int)2e3+5
#define          PI acos(-1.0)
#define          int long long
#define          vi vector<int>
#define          mod 1000000007
#define          ld long double
#define          eb emplace_back
#define          mii map<int,int>
#define          vpii vector<pii>
#define          pii pair<int,int>
#define          pq priority_queue
#define          BLOCK (int)sqrt(N)
#define          test(x) while(x--)
#define          all(x) begin(x),end(x)
#define          allr(x) rbegin(x),rend(x)
#define          fo(i,a,n) for(int i=a;i<n;i++)
#define          rfo(i,n,a) for(int i=n;i>=a;i--)
#define          FAST ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#define          time() cerr << "Time : " << (double)clock() / (double)CLOCKS_PER_SEC << "s\n"
#define 		 bug(...)       __f (#__VA_ARGS__, __VA_ARGS__)

typedef tree< int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update >
OS ;

template <typename Arg1>
void __f (const char* name, Arg1&& arg1) { cout << name << " : " << arg1 << endl; }
template <typename Arg1, typename... Args>
void __f (const char* names, Arg1&& arg1, Args&&... args)
{
     const char* comma = strchr (names + 1, ',');
     cout.write (names, comma - names) << " : " << arg1 << " | "; __f (comma + 1, args...);
}

const int inf = 0x3f3f3f3f;
const int INF = 0x3f3f3f3f3f3f3f3f;

int n,m,k,q;
int a[N] , b[N];
int dp[N][N][6];
int dp1[N][N];

int makeAns(int i = n-1 , int j = m-1 , int rest = k)
{
	if( rest < 0 ) return -inf;
	if( i < 0 )
	{
		if( rest >= ( j + 1) ) return 0;
		else return -inf;
	}
	if( j < 0 )
	{
		if( rest >= ( i + 1) ) return 0;
		else return -inf;
	}

	int& ans = dp[i][j][rest];
	if( ~ans ) return ans;

	ans = 0;

	if( a[i] == b[j] )
	{
		ans = 1 + makeAns(i-1, j-1 , rest);
	}
	else
	{
		int xx = 1 + makeAns(i-1,j-1,rest - 1);
		int yy = makeAns(i-1,j,rest );
		int zz = makeAns(i,j-1,rest);

		ans = max( { xx , yy , zz} );
	}

	return ans;
}

int LIS()
{
	fo(i,1,n+1)
	{
		fo(j,1,m+1)
		{
			dp1[i][j] = max( dp1[i-1][j] , dp1[i][j-1] );
			if( a[i-1] == b[j-1] )
			{
				dp1[i][j] = max( dp1[i][j] , 1 + dp1[i-1][j-1] );
			}

		}
	}

	return dp1[n][m];
}

void go()
{
	cin >> n >> m >> k;

	fo(i,0,n) cin >> a[i];
	fo(i,0,m) cin >> b[i];

	int ans = 0;
	memset( dp , -1 , sizeof dp );

	if( n != m )
	{
		ans = LIS();
		ans += k;
	}
	else
	{
		ans = makeAns();
	}

	cout << ans << endl;
}

int32_t main()
{
	FAST;
	int t=1;
	// cin>>t;
	test(t) go();
}