【CF1189F】Array Beauty【dp】【复杂度玄学优化】
傳送門(mén)
題意:給一個(gè)長(zhǎng)度為NNN的序列aaa,從中選出kkk個(gè),定義一個(gè)序列的美麗度為最接近的兩個(gè)數(shù)的差的絕對(duì)值,求所有方案的美麗度之和模998244353998244353998244353。
N≤1000,ai≤100000N \leq 1000,a_i \leq100000N≤1000,ai?≤100000
顯然先排個(gè)序,枚舉美麗度求方案數(shù)。
設(shè)美麗度為ddd,我們希望求出選出kkk個(gè)數(shù),相鄰兩數(shù)的差不小于ddd
顯然可以dp
dp(i,j)dp(i,j)dp(i,j)表示前iii個(gè)選jjj個(gè)且最后一個(gè)必選的方案數(shù)。
dp(i,j)=∑ai?ak≥ddp(k,j?1)dp(i,j)=\sum_{a_i-a_k\geq d}dp(k,j-1)dp(i,j)=ai??ak?≥d∑?dp(k,j?1)
一個(gè)前綴和就完事了
總復(fù)雜度O(annk)O(a_nnk)O(an?nk)
冷靜分析,我們發(fā)現(xiàn)隨著ddd的增加,數(shù)會(huì)越來(lái)越小。當(dāng)?shù)竭_(dá)一定的程度后就是000了。
容易算出這個(gè)值是?an?a1k?1?\lfloor\frac{a_n-a_1}{k-1}\rfloor?k?1an??a1???
所以只用枚舉這么多個(gè)
復(fù)雜度O(an?a1k?1nk)=O(ann)O(\frac{a_n-a_1}{k-1}nk)=O(a_nn)O(k?1an??a1??nk)=O(an?n)
#include <iostream> #include <cstdio> #include <cstring> #include <cctype> #include <algorithm> #define MAXN 1005 #define MAXM 100005 using namespace std; typedef long long ll; const int MOD=998244353; inline int add(const int& x,const int& y){return x+y>=MOD? x+y-MOD:x+y;} inline int dec(const int& x,const int& y){return x<y? x+MOD-y:x-y;} int a[MAXN]; int dp[MAXN][MAXN],ans[MAXM]; int main() {int n,k;scanf("%d%d",&n,&k);for (int i=1;i<=n;i++) scanf("%d",&a[i]);sort(a+1,a+n+1);int mx=(a[n]-a[1])/(k-1);for (int d=1;d<=mx;d++){for (int i=1;i<=k;i++)for (int j=1;j<=n;j++)dp[i][j]=0;for (int i=1;i<=n;i++) dp[1][i]=1;for (int i=2;i<=k;i++){int pos=0,sum=0;for (int j=1;j<=n;j++){while (pos<n&&a[j]-a[pos+1]>=d)sum=add(sum,dp[i-1][++pos]); dp[i][j]=sum; } }for (int i=1;i<=n;i++) ans[d]=add(ans[d],dp[k][i]);}int res=0;for (int i=1;i<=mx;i++) res=add(res,ans[i]);printf("%d\n",res);return 0; }這題交了四發(fā),是個(gè)憨憨。
總結(jié)
以上是生活随笔為你收集整理的【CF1189F】Array Beauty【dp】【复杂度玄学优化】的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: 下巴自己在动是怎么回事?
- 下一篇: 【CF1230E】Kamil and M