#875141 (C++17) No.2332 Make a Sequence

提出ソース

結果

問題	No.2332 Make a Sequence
ユーザー	👑 potato167
提出日時	2023-05-28 15:47:38
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	TLE
実行時間	-
コード長	4,284 bytes
コンパイル時間	2,682 ms
コンパイル使用メモリ	222,128 KB
実行使用メモリ	24,888 KB
最終ジャッジ日時	2024-06-08 08:35:19
合計ジャッジ時間	63,318 ms
ジャッジサーバーID （参考情報）	judge2 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 10,624 KB
testcase_01	AC	3 ms 5,248 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	2 ms 5,376 KB
testcase_05	AC	2 ms 5,376 KB
testcase_06	AC	1 ms 5,376 KB
testcase_07	AC	994 ms 20,212 KB
testcase_08	AC	26 ms 5,600 KB
testcase_09	AC	515 ms 15,852 KB
testcase_10	AC	1,052 ms 19,280 KB
testcase_11	AC	492 ms 13,732 KB
testcase_12	AC	1,364 ms 23,608 KB
testcase_13	AC	1,366 ms 23,604 KB
testcase_14	AC	1,415 ms 23,608 KB
testcase_15	AC	1,474 ms 23,604 KB
testcase_16	AC	1,394 ms 23,608 KB
testcase_17	AC	1,361 ms 23,608 KB
testcase_18	AC	1,360 ms 23,608 KB
testcase_19	AC	1,409 ms 23,604 KB
testcase_20	AC	1,379 ms 23,608 KB
testcase_21	AC	1,389 ms 23,544 KB
testcase_22	AC	1,366 ms 23,608 KB
testcase_23	AC	1,375 ms 23,608 KB
testcase_24	AC	1,385 ms 23,612 KB
testcase_25	AC	1,374 ms 23,612 KB
testcase_26	AC	1,389 ms 23,604 KB
testcase_27	AC	1,451 ms 23,608 KB
testcase_28	AC	1,455 ms 23,612 KB
testcase_29	AC	1,387 ms 23,484 KB
testcase_30	AC	1,396 ms 23,560 KB
testcase_31	AC	1,427 ms 23,608 KB
testcase_32	AC	1,375 ms 23,608 KB
testcase_33	AC	1,395 ms 23,608 KB
testcase_34	AC	1,395 ms 23,612 KB
testcase_35	AC	1,386 ms 23,608 KB
testcase_36	AC	1,401 ms 23,612 KB
testcase_37	AC	1,365 ms 22,844 KB
testcase_38	AC	1,388 ms 22,968 KB
testcase_39	AC	1,406 ms 22,712 KB
testcase_40	AC	1,373 ms 22,968 KB
testcase_41	AC	1,363 ms 23,224 KB
testcase_42	AC	1,383 ms 20,408 KB
testcase_43	AC	1,406 ms 20,404 KB
testcase_44	AC	1,367 ms 20,924 KB
testcase_45	AC	1,455 ms 21,556 KB
testcase_46	AC	1,406 ms 21,176 KB
testcase_47	TLE	-
testcase_48	--	-
testcase_49	--	-
testcase_50	--	-
testcase_51	--	-
testcase_52	--	-
testcase_53	--	-
testcase_54	--	-
testcase_55	--	-
testcase_56	--	-
testcase_57	--	-
testcase_58	--	-
testcase_59	--	-
testcase_60	--	-
testcase_61	--	-
testcase_62	--	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
using namespace std;
using std::cout;
using std::cin;
using std::endl;
using ll=long long;
using ld=long double;
ll ILL=2167167167167167167;
const int INF=2100000000;
const int mod=998244353;
#define rep(i,a,b) for (ll i=a;i<b;i++)
#define all(p) p.begin(),p.end()
template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>;
template<class T> ll LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> ll UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}else return 0;}
template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}else return 0;}
template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());}
template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});}
void yneos(bool a){if(a) cout<<"Yes\n"; else cout<<"No\n";}
template<class T> void vec_out(vector<T> &p){for(int i=0;i<(int)(p.size());i++){if(i) cout<<" ";cout<<p[i];}cout<<"\n";}
template<class T> T vec_min(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmin(ans,x);return ans;}
template<class T> T vec_max(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmax(ans,x);return ans;}
template<class T> T vec_sum(vector<T> &a){assert(!a.empty());T ans=a[0]-a[0];for(auto &x:a) ans+=x;return ans;}
int pop_count(long long a){int res=0;while(a){res+=(a&1),a>>=1;}return res;}


namespace po167{
//LCP vec[0,n) and vec[i,n)
//for all i 
//O(vec.size())
template<class T>
std::vector<int> Z_algo(std::vector<T> &vec){
	int n=vec.size();
	int ind=1,j=0,k;
	std::vector<int> ans(n,0);
	ans[0]=n;
	while(ind<n){
		while(ind+j<n&&vec[j]==vec[ind+j]) j++;
		ans[ind]=j;
		if(j==0){
			ind++;
			continue;
		}
		k=1;
		while(k+ind<n&&ans[k]+k<j){
			ans[k+ind]=ans[k];
			k++;
		}
		j-=k;
		ind+=k;
	}
	return ans;
}
std::vector<int> Z_algo(std::string &vec){
	int n=vec.size();
	int ind=1,j=0,k;
	std::vector<int> ans(n,0);
	ans[0]=n;
	while(ind<n){
		while(ind+j<n&&vec[j]==vec[ind+j]) j++;
		ans[ind]=j;
		if(j==0){
			ind++;
			continue;
		}
		k=1;
		while(k+ind<n&&ans[k]+k<j){
			ans[k+ind]=ans[k];
			k++;
		}
		j-=k;
		ind+=k;
	}
	return ans;
}
}
using po167::Z_algo;

void solve();
// oddloop
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	
	int t=1;
	//cin>>t;
	rep(i,0,t) solve();
}

void solve(){
	ll N,M;
	cin>>N>>M;
	vector<int> A(N),B(M);
	vector<ll> C(M+1);
	rep(i,0,N) cin>>A[i];
	rep(i,0,M) cin>>B[i];
	rep(i,0,M) cin>>C[i];
	auto p=A;
	p.push_back(-1);
	rep(i,0,M) p.push_back(B[i]);
	auto tmp=Z_algo(p);
	vector<int> Z(M);
	rep(i,0,M) Z[i]=tmp[N+i+1];
	vector<vector<int>> H(M+1);
	rep(i,0,M) H[i+Z[i]].push_back(i);
	vector<ll> dp(M+1,ILL);
	vector<bool> ok(M+1);
	vector<int> order(M);
	rep(i,0,M) order[i]=i;
	sort(all(order),[&](int l,int r){
		return C[l]>C[r];
	});
	dp[0]=0;
	ok[0]=1;
	int Q=200;
	vector<int> ch;
	rep(i,1,M+1){
		if((i-1)%Q==0){
			int L=i,R=min(i+Q,M+1);
			ch.clear();
			vector<int> lis;
			rep(j,0,M){
				int a=order[j];
				if(L<=a&&a<R) ch.push_back(a);
				else if(ok[a]){
					if(a+Z[a]<R) ch.push_back(a);
					else{
						if(!lis.empty()){
							int ind=lis.size();
							ind-=1;
							if(C[lis[ind]]==C[a]){
								if(dp[lis[ind]]>C[a]) lis[ind]=a;
							}
							else lis.push_back(a);
						}
						else lis.push_back(a);
					}
				}
			}
			vector<int> st;
			int k=-1;
			// (a/b)<=(c/d) ?
			auto f=[](ll a,ll b,ll c,ll d)->bool{
				if(a/b!=c/d) return a/b<c/d;
				a%=b;
				c%=d;
				return a*d<=c*b;
			};
			for(auto c:lis){
				while(1<=k){
					int a=st[k-1];
					int b=st[k];
					if(f(dp[c]-dp[a],C[a]-C[c],dp[b]-dp[a],C[b]-C[c])){
						k--;
					}
					else break;
				}
				k++;
				st.push_back(c);
			}
			int ind=0;
			if(k!=-1){
				rep(j,L,R){
					while(ind!=k){
						int a=st[ind];
						int b=st[ind+1];
						if(dp[a]+j*C[a]<=dp[b]+j*C[b]) break;
						ind++;
					}
					dp[j]=dp[st[ind]]+j*C[st[ind]];
				}
			}
		}
		for(auto x:ch){
			if(ok[x]){
				chmin(dp[i],dp[x]+C[x]*i);
			}
		}
		//cout<<i<<" "<<dp[i]<<endl;
		dp[i]-=i*C[i];
		ok[i]=1;
		for(auto z:H[i]) ok[z]=0;
	}
	cout<<(dp[M]==ILL?-1:dp[M])<<"\n";
}

yukicoder

結果

テストケース

ソースコード