【CF1189F】Array Beauty【dp】【复杂度玄学优化】
傳送門
題意:給一個長度為NNN的序列aaa,從中選出kkk個,定義一個序列的美麗度為最接近的兩個數的差的絕對值,求所有方案的美麗度之和模998244353998244353998244353。
N≤1000,ai≤100000N \leq 1000,a_i \leq100000N≤1000,ai?≤100000
顯然先排個序,枚舉美麗度求方案數。
設美麗度為ddd,我們希望求出選出kkk個數,相鄰兩數的差不小于ddd
顯然可以dp
dp(i,j)dp(i,j)dp(i,j)表示前iii個選jjj個且最后一個必選的方案數。
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)
一個前綴和就完事了
總復雜度O(annk)O(a_nnk)O(an?nk)
冷靜分析,我們發現隨著ddd的增加,數會越來越小。當到達一定的程度后就是000了。
容易算出這個值是?an?a1k?1?\lfloor\frac{a_n-a_1}{k-1}\rfloor?k?1an??a1???
所以只用枚舉這么多個
復雜度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; }這題交了四發,是個憨憨。
總結
以上是生活随笔為你收集整理的【CF1189F】Array Beauty【dp】【复杂度玄学优化】的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 下巴自己在动是怎么回事?
- 下一篇: 夜晚咽喉疼痛怎么回事?