結果

問題 No.3097 Azuki Kurai
ユーザー kotatsugame
提出日時 2025-04-06 16:48:12
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 847 ms / 4,000 ms
コード長 1,030 bytes
コンパイル時間 1,135 ms
コンパイル使用メモリ 83,136 KB
実行使用メモリ 9,436 KB
最終ジャッジ日時 2025-04-06 16:48:23
合計ジャッジ時間 10,024 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<cassert>
#include<atcoder/maxflow>
using namespace std;
int N,M,K;
int A[9],B[2020];
long solve(int i)
{
	atcoder::mf_graph<long>mf((i+1)*2*N+N+2);
	int S=(i+1)*2*N+N;
	int T=S+1;
	for(int j=0;j<N;j++)
	{
		mf.add_edge(S,j,A[j]);
		for(int t=0;t<=i;t++)
		{
			mf.add_edge(t*2*N+j,t*2*N+j+N,K);
			if(B[t]!=(j+1)%N)mf.add_edge(t*2*N+j+N,(t+1)*2*N+(j+1)%N,K);
			if(B[t]!=(j+N-1)%N)mf.add_edge(t*2*N+j+N,(t+1)*2*N+(j+N-1)%N,K);
			if(B[t]!=j)mf.add_edge(t*2*N+j,(t+1)*2*N+j,(long)1e10);
		}
		mf.add_edge((i+1)*2*N+j,T,(long)1e10);
	}
	return mf.flow(S,T);
}
long ans[2020];
void dfs(int l,int r)
{
	if(r-l<=1)return;
	if(ans[l]==ans[r])
	{
		for(int k=l+1;k<r;k++)ans[k]=ans[l];
		return;
	}
	int m=(l+r)/2;
	ans[m]=solve(m);
	dfs(l,m);
	dfs(m,r);
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>N>>M>>K;
	for(int i=0;i<N;i++)cin>>A[i];
	for(int i=0;i<M;i++)cin>>B[i],B[i]--;
	ans[0]=solve(0);
	if(M>=2)ans[M-1]=solve(M-1);
	dfs(0,M-1);
	for(int i=0;i<M;i++)cout<<ans[i]<<"\n";
}
0