Submission #633798

#	Time	Username	Problem	Language	Result	Execution time	Memory
633798		IWTIM	Copy and Paste 3 (JOI22_copypaste3)	C++14	100 / 100	477 ms	69580 KiB

copypaste3

# include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define f first
#define s second
#define int long long
#define pii pair <int, int>
const int N = 3005, mod = 1e9 + 7;
int t,n,a,b,c,nxt[N], dp[N][N],pw[N],p;
vector < pii > v;
int hs[N][N];
string s;
int get(char ch) {
	return (ch - 'a' + 1);
}
void go(vector <int> vv, int sz) {
	int cur = -1;
	for (int i = (int)vv.size() - 1; i >= 0; i--) {
		while(true) {
			if (vv.back() >= vv[i] + sz) {
				cur = vv.back();
				vv.pop_back();
			} else break;
		}
		nxt[vv[i]] = cur;
	}
}
main() {
	p = 37;
    cin>>n>>s>>a>>b>>c;
    s = "@" + s;
    for (int i = 1; i <= n; i++) {
    	for (int j = i; j <= n; j++) {
    		dp[i][j] = (j - i + 1)*a;
		}
	}
	pw[0] = 1;
	for (int i = 1; i <= n; i++) {
		pw[i] = (pw[i - 1] * p) % mod;
	}
	for (int i = 1; i <= n; i++) {
		int curhs = 0;
		for (int j = i; j <= n; j++) {
			curhs = (curhs * p + get(s[j])) % mod;  
			hs[i][j] = curhs;
		}
	}
	for (int sz = 1; sz <= n; sz++) {
		v.clear();
		for (int j = 1; j + sz - 1 <= n; j++) {
			v.pb({hs[j][j + sz - 1], j});
		}
		
		sort(v.begin(),v.end());
		v.pb({-1, 0});
		int le = 0;
		for (int i = 1; i < v.size(); i++) {
			if (v[i].f != v[i - 1].f) {
				int ri = i - 1;
				vector <int> vv;
				for (int j = le; j <= ri; j++) {
					vv.pb(v[j].s);
				}
				go(vv,sz);
				le = i;
			}
		}
		
		for (int l = 1; l + sz - 1 <= n; l++) {
			if (sz > 1) dp[l][l + sz - 1] = min(dp[l][l + sz - 1], min(dp[l + 1][l + sz - 1] + a, dp[l][l + sz - 2] + a)); 
		}
		for (int l = 1; l + sz - 1 <= n; l++) {
			int curl = l;
			int cnt = 1;
			while (nxt[curl] != -1) {
				cnt++;
				int ri = nxt[curl] + sz - 1;
				dp[l][nxt[curl] + sz - 1] = min(dp[l][nxt[curl] + sz - 1], dp[l][l + sz - 1] + b + cnt * c + (ri - l + 1 - sz * cnt)*a);
			    curl = nxt[curl];
			}
		}
	}
	cout<<dp[1][n]<<"\n";
}

Compilation message (stderr)

copypaste3.cpp:28:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   28 | main() {
      | ^~~~
copypaste3.cpp: In function 'int main()':
copypaste3.cpp:57:21: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   57 |   for (int i = 1; i < v.size(); i++) {
      |                   ~~^~~~~~~~~~

• Subtask 1: 1 / 1
• Subtask 2: 5 / 5
• Subtask 3: 14 / 14
• Subtask 4: 10 / 10
• Subtask 5: 32 / 32
• Subtask 6: 38 / 38