Codeforces Round #522 div 2 C. Playing Piano dp
阿新 • • 發佈:2018-11-22
題意:給你一個a陣列,你要按要求構造等長的b陣列,如果a[ i ]>a[ i-1 ],那麼b[ i ]>b[ i-1 ],若a[ i ]<a[ i-1],那麼b[ i ]<b[ i-1 ],若a[ i ]=a[ i-1 ],那麼b[ i ]!=b[ i-1 ],且b陣列最大不超過5,最小為1。
思路:一個很有意思的題,設d[ i ][ j ][ k ]表示b[ i ]=j 且b[ i-1 ]=k時,是否合理,狀態轉移很簡單,初始化d[ 1 ][ j ][ k ]全為1,若a[ i ]>a[ i-1 ],那麼d[ i ][ j ][ k ] |= d[ i-1 ][ k ][ p ](j>k,1<=p<=5),就ok了(歡迎大家優化這個dp)。
#include<cstdio> #include<cstring> #include<stack> using namespace std; const int maxn=1e5+10; int d[maxn][6][6],a[maxn]; stack<int>s; int main() { int n,x; scanf("%d%d",&n,&a[1]); if(n==1) { puts("1"); return 0; } memset(d[1],1,sizeof(d[1])); for(int i=2;i<=n;i++) { scanf("%d",&a[i]); for(int j=1;j<=5;j++) for(int k=1;k<=5;k++) for(int p=1;p<=5;p++) { if(a[i]==a[i-1]) { if(j!=k) d[i][j][k]|=d[i-1][k][p]; } else if(a[i]>a[i-1]) { if(j>k) d[i][j][k]|=d[i-1][k][p]; } else if(a[i]<a[i-1]) { if(j<k) d[i][j][k]|=d[i-1][k][p]; } } } int p=0; for(int j=1;j<=5;j++) for(int k=1;k<=5;k++) if(d[n][j][k])p=j; if(!p)puts("-1"); else { for(int i=n;i;i--) { s.push(p); for(int j=1;j<=5;j++) if(d[i][p][j]) { p=j; break; } } while(!s.empty()) printf("%d ",s.top()),s.pop(); } }