题目大意:
有一排n个格子和2枚硬币。
现在有q次任务,每一次要你把其中一枚硬币移到x的位置上,移动1格的代价是1。
两枚硬币不能同时移动,任务必须按次序完成。
现在告诉你两枚硬币初始状态所在的位置a和b,问完成所有任务的最小代价。
思路:
很容易想到一个O(qn)的DP。
由于完成任务的次序确定,每个任务的位置也确定,我们可以用f[i][j]表示完成第i个任务后,一个硬币在x[i],一个硬币在j的最小代价。
转移方程为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的测试数据。
不难发现,在状态转移方程中,如果我们能去掉绝对值,里面的东西就能用线段树维护。
而绝对值的取值只与硬币的左右位置关系有关。
因此我们可以建2棵线段树,一棵表示被转移的状态在目标状态左边,一棵表示在右边。
左线段树中每个叶子结点x[i-1]维护f[i-1][j]-x[i-1]的值,右线段树每个叶子结点x[i-1]维护f[i-1][j]+x[i-1]的值。
看了一下榜,发现排在前面的基本上都是用树状数组做的。
然而用树状数组维护区间最值难道不是O(log^2 n)的吗?
事实上我们可以发现线段树上维护的东西只会越来越小,这样我们可以直接在树状数组上修改,不用考虑原来的最小值没了怎么办。
然后我又在树状数组里面加了一个剪枝。
这样随随便便就能拿Rank1。
1 #include<cstdio> 2 #include<cctype> 3 #include<cstdlib> 4 #include<algorithm> 5 typedef signed long long int int64; 6 inline unsigned getint() { 7 register char ch; 8 while(!isdigit(ch=getchar())); 9 register unsigned x=ch^'0'; 10 while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0'); 11 return x; 12 } 13 inline int64 min(const int64 &a,const int64 &b) { 14 return a<b?a:b; 15 } 16 const int64 inf=0x7ffffffffffffff; 17 const int N=200001; 18 int n; 19 class FenwickTree { 20 private: 21 int64 val[N]; 22 int lowbit(const int &x) const { 23 return x&-x; 24 } 25 public: 26 FenwickTree() { 27 std::fill(&val[0],&val[N],inf); 28 } 29 void modify(int p,const int64 &x) { 30 while(p<=n) { 31 if(x<val[p]) { 32 val[p]=x; 33 } else { 34 return; 35 } 36 p+=lowbit(p); 37 } 38 } 39 int64 query(int p) const { 40 int64 ret=inf; 41 while(p) { 42 ret=min(ret,val[p]); 43 p-=lowbit(p); 44 } 45 return ret; 46 } 47 }; 48 FenwickTree ta; 49 class RevFenwickTree { 50 private: 51 int64 val[N]; 52 int lowbit(const int &x) const { 53 return x&-x; 54 } 55 public: 56 RevFenwickTree() { 57 std::fill(&val[0],&val[N],inf); 58 } 59 void modify(int p,const int64 &x) { 60 while(p) { 61 if(x<val[p]) { 62 val[p]=x; 63 } else { 64 return; 65 } 66 p-=lowbit(p); 67 } 68 } 69 int64 query(int p) const { 70 int64 ret=inf; 71 while(p<=n) { 72 ret=min(ret,val[p]); 73 p+=lowbit(p); 74 } 75 return ret; 76 } 77 }; 78 RevFenwickTree tb; 79 int64 f[N]; 80 inline void modify(const int &p,const int64 x) { 81 if(x<f[p]) { 82 f[p]=x; 83 ta.modify(p,x-p); 84 tb.modify(p,x+p); 85 } 86 } 87 int main() { 88 n=getint(); 89 int q=getint(),a=getint(),b=getint(); 90 std::fill(&f[0],&f[N],inf); 91 modify(a,0); 92 int64 sum=0; 93 while(q--) { 94 a=b; 95 b=getint(); 96 sum+=abs(a-b); 97 int64 t1=ta.query(b)+b,t2=tb.query(b)-b; 98 modify(a,min(t1,t2)-abs(a-b)); 99 } 100 int64 tmp=inf; 101 for(register int i=1;i<=n;i++) { 102 tmp=min(tmp,f[i]); 103 } 104 printf("%lld\n",tmp+sum); 105 return 0; 106 }
原来的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 }