題目描述:
/**
【編碼題】
字符串S由小寫字母構成,長度為n。定義一種操作,
每次都可以挑選字符串中任意的兩個相鄰字母進行交換。
詢問在至多交換m次之后,字符串中最多有多少個連續的位置上的字母相同?
輸入描述:
第一行為一個字符串S與一個非負整數m。(1 <= |S| <= 1000, 1 <= m <= 1000000)
輸出描述:
一個非負整數,表示操作之后,連續最長的相同字母數量。
輸入例子1:
abcbaa 2
輸出例子1:
2
例子說明1:
使2個字母a連續出現,至少需要3次操作。即把第1個位置上的a移動到第4個位置。
所以在至多操作2次的情況下,最多只能使2個b或2個a連續出現。
*/
思路如下:
維護一個表每個
每行存放同樣字符之間的位置(按原來順序擺放)
在位置向量中
dp[i][j] = dp[i+1][j-1] + abs(posVec[j] - posVec[i]) - (j-i);
dp[i][j]第i個數->第j個數移動的次數
base case:
dp[i][i]=0
dp[i][i+1]= abs(posVec[i+1] - posVec[i]) - 1
i<=j
這里只枚舉i<=j是因為 j<=i時候其實是對稱的情況
代碼如下:
#include<stdio.h>
#include<iostream>
#include<vector>
#define MAX_ROW 26
using namespace std;
vector<int> posTable[MAX_ROW];
int DP(const vector<int> &posVec, int K)
{
int n = posVec.size();
vector< vector<int> > dp(n, vector<int>(n, 0));
for(int i=0; i<n-1; ++i)
{
dp[i][i+1] = abs(posVec[i+1] - posVec[i]) - 1;
}
for(int j=2; j<n; ++j)
{
for(int i=0; i<n-j; ++i)
{
int row = i, col = i+j;
dp[row][col] = dp[row+1][col-1] + abs(posVec[col] - posVec[row]) - (col-row);
}
}
int Max = 0;
for(int i=0; i<n; ++i)
{
for(int j=i; j<n; ++j)
{
if (dp[i][j] <= K)
{
Max = max(Max, j-i+1);
}
}
}
return Max;
}
int main()
{
//建立posTable;
string line;
int K;
cin>>line>>K;
for(int i=0; line[i]!='\0'; i++)
{
if(line[i]<'a' || line[i]>'z')
return -1;
posTable[line[i]-'a'].push_back(i);
}
//找最大滑動窗口
int maxSize=0;
for(int i=0; i<MAX_ROW; i++)
{
vector<int> &pos=posTable[i];
maxSize=max(maxSize, DP(pos, K));
}
printf("%d", maxSize);
return 0;
}