有一串玛瑙项链,项链上面有N个玛瑙珠子,这些玛瑙有M (M < = 9) 种不同的颜色,购买的时候可以只选择购买项链长度为K(K<=n)的一段。现在有个苦逼的找不到工作的程序员想花最少的钱买到颜色最多的一段项链。要求时间复杂度O(N),空间复杂度O(1)。
答案:看到这道题,第一时间想不到O(N)的思路不要慌,先来看看最裸的思路应该怎么写?枚举每一个买项链的位置,然后模拟一遍统计有多少个不同的珠子,代码是这样的:
-
<span style="font-size:14px;">for (int i=0;i<n;i++) nek[i+n]=nek[i]; //首先把项链的长度扩大一倍,这样就把环拆成了一个链
-
for (int i=0;i<n;i++) //枚举每一个开头
-
{
-
int vis[10]; //vis是用来统计每一个颜色的珠子出现的次数
-
memset(vis,0,sizeof(vis));
-
for (int j=i;j<i+k;j++)
-
{
-
vis[arr[j]]++;
-
}
-
int num=0;
-
for (int j=1;j<=9;j++)
-
if (vis[j]>0)
-
num++;
-
ans=max(ans,num);
-
}</span>
现在这样做,复杂度是O(N*K)的,当K等于N的时候,复杂度就退化成了O(N*N)。所以要考虑如何降低一维复杂度,题目要求复杂度是O(N),因此,里面的K这个复杂度是有可能被降下来的。来考虑下面一个例子:
1 1 2 3 4
假设K=4,那么[1,4]这个区间里面,vis数组的值是:2 1 1 0,而[2,5]这个区间里面,vis数组的值是[1,1,1,1],这个是怎么得到的呢?在[1,4]的vis的基础上,vis[1]--,vis[4]++,只进行了两次操作,就可以得到[2,5]区间的vis情况。所以,我们可以在一开始加入K个珠子进来,然后,每次,把最先进来的那个珠子X踢出去,此时vis[x]--,然后,再把下一个珠子Y加进来,此时vis[y]++,仅仅两次操作,就可以得到下一个位置的值了。代码写起来也更方便:
-
<span style="font-size:14px;">for (int i=0;i<n;i++) nek[i+n]=nek[i]; //首先把项链的长度扩大一倍,这样就把环拆成了一个链
-
int vis[10]; //vis是用来统计每一个颜色的珠子出现的次数
-
memset(vis,0,sizeof(vis));
-
int num=0;
-
for (int i=0;i<k-1;i++) //首先把前k-1个珠子加进来
-
{
-
vis[arr[i]]++;
-
if (vis[arr[i]]==1) //如果这个珠子是第一次出现,num++
-
num++;
-
}
-
for (int i=k-1;i<n+k;i++) //枚举将每一个珠子i加进来
-
{
-
//首先加入第i颗珠子
-
vis[arr[i]]++;
-
if (vis[arr[i]]==1)
-
num++;
-
ans=max(ans,num);
-
//然后把当前第一颗珠子删掉
-
vis[arr[i-(k-1)]]--;
-
if (vis[arr[i-(k-1)]]==0) //如果这颗珠子是自己颜色下唯一的那一颗
-
num--;
-
}</span>
如此一来,时间复杂度从O(N*K)降低到了O(N),空间复杂度是O(10)=O(1)