srm465 div2 1000
题意是有一段连续的时刻表,有n个时刻,现在给你的仅仅是分钟序列,让你复原小时序列,并求出按字典序排列第k小的序列。
我用f[i][j]表示第i位为j小时的序列有多少种。
首先如果我已经求出了所有的f[i][j],那么为了求第k小的,我会这样做:
从f[0][0]开始枚举,看k,如果k>f[0][0],那么让k-=f[0][0],继续找f[0][1]。。。。这样一直找直到找到k<=f[0][j],那么第一位就确定了是j;
再以j为基础找下一位。。。。一直缩小范围直到找到为止。
现在的问题就是怎么求f[i][j];
我会初始化f[n-1][0]------f[n-1][23]为1;
转移方程: m[i-1]>=m[i] f[i-1][j] += f[i][j+1]-------f[i][23];如果前面分钟大于等于后面分钟,那么前面的j小时只可能由后面 j+1小时 到 23小时推得
m[i-1]<m[i] f[i-1][j] +=f[i][j]-----------f[i][23];
vector <int> WeirdTimes::hourValues(vector <int> minuteValues, int K) {
int n = minuteValues.size();
long long f[55][24];
memset(f,0,sizeof(f));
for(int i = 0; i < 24; i++)f[n-1][i]=1;
for(int i = n-1; i >= 0; i--)
{
for(int j = 23; j >= 0; j--)
{
if(i-1>=0&&minuteValues[i]>minuteValues[i-1])
{
for(int k = j; k < 24; k++) f[i-1][j]+=f[i][k];
}
else if(i-1>=0&&minuteValues[i]<=minuteValues[i-1])
{
for(int k = j+1; k < 24; k++) f[i-1][j]+=f[i][k];
}
}
}
// for(int i = 0; i < 24; i++)printf("%d=%d\n",i,f[0][i]);
vector <int> a;
long long sum = (long long)K;
for(int i = 0; i < 24; i++)sum-=f[0][i];
if(sum>(long long)0)
{
a.push_back(-1);return a;
}
int cur = 0;
for(int i = 0; i < n; i++)
{
for(int j=cur; j < 24; j++)
{
if((long long)K>f[i][j])K-=f[i][j];
else
{
a.push_back(j);
cur = j;
if(i+1<n&&minuteValues[i+1]<=minuteValues[i])cur++;
break;
}
}
}
return a;
}