[AtCoder-ARC073F]Many Moves
題目大意:
有一排n個格子和2枚硬幣。
現(xiàn)在有q次任務,每一次要你把其中一枚硬幣移到x的位置上,移動1格的代價是1。
兩枚硬幣不能同時移動,任務必須按次序完成。
現(xiàn)在告訴你兩枚硬幣初始狀態(tài)所在的位置a和b,問完成所有任務的最小代價。
思路:
很容易想到一個O(qn)的DP。
由于完成任務的次序確定,每個任務的位置也確定,我們可以用f[i][j]表示完成第i個任務后,一個硬幣在x[i],一個硬幣在j的最小代價。
轉(zhuǎn)移方程為f[i][j]=min{f[i-1][j]+|x[i]-x[i-1]|},f[i][a[i-1]]=min{f[i-1][j]+|x[i]-j|}。
然而這樣還是會TLE,在AtCoder上只過了14/34的測試數(shù)據(jù)。
不難發(fā)現(xiàn),在狀態(tài)轉(zhuǎn)移方程中,如果我們能去掉絕對值,里面的東西就能用線段樹維護。
而絕對值的取值只與硬幣的左右位置關(guān)系有關(guān)。
因此我們可以建2棵線段樹,一棵表示被轉(zhuǎn)移的狀態(tài)在目標狀態(tài)左邊,一棵表示在右邊。
左線段樹中每個葉子結(jié)點x[i-1]維護f[i-1][j]-x[i-1]的值,右線段樹每個葉子結(jié)點x[i-1]維護f[i-1][j]+x[i-1]的值。
看了一下榜,發(fā)現(xiàn)排在前面的基本上都是用樹狀數(shù)組做的。
然而用樹狀數(shù)組維護區(qū)間最值難道不是O(log^2 n)的嗎?
事實上我們可以發(fā)現(xiàn)線段樹上維護的東西只會越來越小,這樣我們可以直接在樹狀數(shù)組上修改,不用考慮原來的最小值沒了怎么辦。
然后我又在樹狀數(shù)組里面加了一個剪枝。
這樣隨隨便便就能拿Rank1。
原來的O(n^2)DP程序:
1 #include<cstdio> 2 #include<cctype> 3 #include<cstring> 4 #include<cstdlib> 5 inline unsigned getint() { 6 register char ch; 7 while(!isdigit(ch=getchar())); 8 register unsigned x=ch^'0'; 9 while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0'); 10 return x; 11 } 12 inline unsigned min(const unsigned &a,const unsigned &b) { 13 return a<b?a:b; 14 } 15 const unsigned N=200000; 16 unsigned long long f[2][N]; 17 unsigned a[2]; 18 int main() { 19 unsigned n=getint(),q=getint(); 20 memset(f[0],0xff,n<<3); 21 a[0]=getint()-1,f[0][getint()-1]=0; 22 for(register unsigned i=1;i<=q;i++) { 23 a[i&1]=getint()-1; 24 memset(f[i&1],0xff,n<<3); 25 for(register unsigned j=0;j<n;j++) { 26 if(!~f[~i&1][j]) continue; 27 f[i&1][j]=min(f[i&1][j],f[~i&1][j]+abs(a[i&1]-a[~i&1])); 28 f[i&1][a[~i&1]]=min(f[i&1][a[~i&1]],f[~i&1][j]+abs(a[i&1]-j)); 29 } 30 } 31 unsigned long long ans=~0; 32 for(register unsigned i=0;i<n;i++) { 33 ans=min(ans,f[q&1][i]); 34 } 35 printf("%llu\n",ans); 36 return 0; 37 } View Code?
轉(zhuǎn)載于:https://www.cnblogs.com/skylee03/p/7609824.html
總結(jié)
以上是生活随笔為你收集整理的[AtCoder-ARC073F]Many Moves的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 移动游戏加载性能和内存管理全解析 学习
- 下一篇: 详解摘要认证