木棍分割

wlesq發表於2024-04-01

[HAOI2008] 木棍分割

題目描述

有n根木棍, 第i根木棍的長度為Li,n根木棍依次連結了一起, 總共有n-1個連線處. 現在允許你最多砍斷m個連線處, 砍完後n根木棍被分成了很多段,要求滿足總長度最大的一段長度最小, 並且輸出有多少種砍的方法使得總長度最大的一段長度最小. 並將結果mod 10007。。。

輸入格式

輸入檔案第一行有2個數n,m. 接下來n行每行一個正整數Li,表示第i根木棍的長度.

輸出格式

輸出有2個數, 第一個數是總長度最大的一段的長度最小值, 第二個數是有多少種砍的方法使得滿足條件.

樣例 #1

樣例輸入 #1

3 2                           
1 
1
10

樣例輸出 #1

10 2

提示

兩種砍的方法: (1)(1)(10)和(1 1)(10)

資料範圍

n<=50000, 0<=m<=min(n-1,1000).

1<=Li<=1000.

我們很容易可以看出用二分求出長度最小值,然後求方案數
首先如果用DFS

點選檢視程式碼
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N =5e4+10,mod=10007;
int n,m,a[N],sum[N];
bool cut[N],vis[N];
bool check(ll mid)
{
	ll tot=0,num=0;ll pre=0;
	for(int i=1;i<=n;i++)
	{
		if(num>m)return 0;
		if(pre+a[i]<=mid)pre+=a[i];
		else 
		{
			tot=max(tot,pre);
			pre=a[i];
			num++;
		}
	}
	tot=max(tot,pre);
	if(tot>mid||num>m)return 0;
	return 1;
}
ll work;ll len;
void dfs(int now,int cnt,int maxle,int pre)
{
//	cout<<cnt<<" "<<maxle<<endl;
	if(maxle>len||cnt>m)return;
	if(cnt<=m)
	{
//		cout<<pre<<endl;
		if(maxle==len||sum[n]-sum[pre-1]==len)work++;
//		cout<<"%%"<<maxle<<endl;
		if(cnt==m)return ;
	}
	for(int i=now;i<=n;i++)
	{
		if(!cut[i])
		{
			cut[i]=1;
//			cout<<cnt<<" "<<i<<" "<<pre<<endl;
			dfs(i+1,cnt+1,max(maxle,sum[i]-sum[pre-1]),i+1);
			cut[i]=0;
		}
	}
}
void clear()
{
	for(int i=1;i<=n;i++)
	{
		vis[i]=cut[i]=0;
	}
}

void fen(ll l,ll r)
{
	while(l<=r)
	{
		ll mid=(l+r)>>1;
		clear();
		if(check(mid))
		{
			r=mid-1;
			len=mid;
		}
		else
		{
			l=mid+1;
		}
	}
}
int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	cin>>n>>m;
	int l=0;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		l=max(l,a[i]);
		sum[i]=sum[i-1]+a[i];
	}
	fen(l,sum[n]);
	dfs(2,0,a[1],1);
	cout<<len<<" "<<work<<endl;
	return 0;
}
/*
3 1
2 4 3
*/

但是回超時樂

image

所以改變思路,用DP
如果\(f[i,j]\)表示前j個木棍分成i組
這裡\(O(n^3)\)顯然不行
所以我們可以記錄一下從不同下標木棍開始最多到哪裡<=len//這裡有點像容斥(就是總方案數減去不合法方案數)。
這樣就不用每次再從1開始找
然後用S維護字首和,滾動陣列否則炸記憶體

image

點選檢視程式碼
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N =5e4+10,mod=10007;
int n,m,a[N],sum[N];
bool cut[N],vis[N];
bool check(ll mid)
{
	ll tot=0,num=0;ll pre=0;
	for(int i=1;i<=n;i++)
	{
		if(num>m)return 0;
		if(pre+a[i]<=mid)pre+=a[i];
		else 
		{
			tot=max(tot,pre);
			pre=a[i];
			num++;
		}
	}
	tot=max(tot,pre);
	if(tot>mid||num>m)return 0;
	return 1;
}
ll len;
int f[N],rem[N],S[N];
int dp()
{
	int k=0;
	for(int i=1;i<=n;i++)
		for(;k<i;k++)
		{
			if(sum[i]-sum[k]<=len)
			{
				rem[i]=k;
				break;
			}
		}
	int res=(sum[n]<=len);
    for(int i=1;i<=n;++i)
    {
    	if(sum[i]<=len) f[i]=1;
    	S[i]=(S[i-1]+f[i])%mod;
    }
	for(int i=2;i<=m+1;i++)
	{
		for(int j=1;j<=n;j++)
		{
			f[j]=S[j-1];
			if(rem[j]-1>=0)
			{
				f[j]=((f[j]-S[rem[j]-1])%mod+mod)%mod;
			}
		}
		for(int j=1;j<=n;j++)
			S[j]=(S[j-1]+f[j])%mod;
		res=(res+f[n])%mod;
	}
	return res;
//	cout<<f[n]<<endl;
}
void fen(ll l,ll r)
{
	while(l<=r)
	{
		ll mid=(l+r)>>1;
		if(check(mid))
		{
			r=mid-1;
			len=mid;
		}
		else
		{
			l=mid+1;
		}
	}
}
int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	cin>>n>>m;
	int l=0;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		l=max(l,a[i]);
		sum[i]=sum[i-1]+a[i];
	}
	fen(l,sum[n]);
	cout<<len<<" "<<dp()<<endl;
	return 0;
}
/*
3 1
2 4 3
*/

相關文章