微軟大樓設計方案(困難)
題目鏈接:https://nanti.jisuanke.com/t/15773
近日,微軟新大樓的設計方案正在廣泛征集中,其中一種方案格外引人註目。在這個方案中,大樓由 nn 棟樓組成,這些樓從左至右連成一排,編號依次為 11 到 nn,其中第 ii 棟樓有 h_ih?i?? 層。每棟樓的每一層為一個獨立的 辦公區域,可以步行 直達同層相鄰樓棟的辦公區域,以及 直達同樓棟相鄰樓層的辦公區域。
由於方案設計巧妙,上一層樓、下一層樓、向左右移動到相鄰樓棟同層的辦公區域均剛好需要 11 分鐘。在這些辦公區域中,有一些被 核心部門 占用了(一個辦公區域內最多只有一個核心部門),出於工作效率的考慮,微軟希望核心部門之間的移動時間越短越好。對於一個給定的 最大移動時間 kk,大樓的 協同值 定義為:有多少個 核心部門對 之間的移動時間不超過 kk。由於大樓門禁的限制,不可以走出整個大樓,也不可以登上天臺思考人生。你可以認為在辦公區域內的移動時間忽略不計,並且在大樓內總是按照最優方案進行移動。
對於一個給定的新大樓設計方案,你能算出方案的協同值麽?
輸入格式
第一行包含兩個正整數 n,k(1\leq k\leq 200020)n,k(1≤k≤200020),分別表示大樓的棟數以及最大移動時間。
第二行包含 nn 個正整數 h_1,h_2,...,h_n(1\leq h_i\leq 20)h?1??,h?2??,...,h?n??(1≤h?i??≤20),分別表示每棟樓的層數。
接下來一行包含一個正整數 mm,表示 核心部門 個數。
接下來 mm 行,每行兩個正整數 x_i,y_i(1\leq x_i\leq n,1\leq y_i\leq h_{x_i})x?i??,y?i??(1≤x?i??≤n,1≤y?i??≤h?x?i????),表示該核心部門位於第 x_ix?i?? 棟樓的第 y_iy?i?? 層。
輸入數據保證 mm 個核心部門的位置不會重復。
對於簡單版本:1\leq n,m\leq 501≤n,m≤50;
對於中等版本:1\leq n\leq 200000,1\leq m\leq 20001≤n≤200000,1≤m≤2000;
對於困難版本:1\leq n,m\leq 2000001≤n,m≤200000。
輸出格式
輸出一個整數,即整個大樓的 協同值。
樣例解釋
樣例對應題目描述中的圖,核心部門 11 和核心部門 33 之間的距離為 8>78>7,因此不能計入答案。
樣例輸入
5 7 4 1 1 3 1 3 1 4 3 1 4 3
樣例輸出
2今天必須過掉,等過了以後填坑。 2017.6.6 過掉之後居然忘了填坑……現在來填 其實有一種很巧妙的方法,水平距離<=k-38的,必定可以到達;水平距離大於k的,必定無法到達。那麽我們要找與(xi,yi)節點滿足距離<=k的點的個數,不妨假設同一列的只考慮比它靠下的點,不同列的只考慮該點左邊的點。那麽我們要考慮的復雜的區間其實只有[ xi-k , xi-(k-38) )這一段。對於這一段區間,可以直接枚舉點,求距離。 現在分析一下復雜度:對於每個點,同一列的統計O(20);水平距離小於等於k-38的區間,維護前綴和,O(1)查詢;剩下那一段小區間,至多37*20個點,每個點可以O(20)來計算距離。總體復雜度是O(n*37*20*20)。 代碼:
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=200005;
vector<int> G[maxn];
int x[maxn],y[maxn];
int cou[maxn];
int val[maxn];
int tj[maxn][25];
int querymi(int l,int r)
{
for (int i=1;i<=20;i++)
{
if (tj[r][i]-tj[l-1][i]>0) return i;
}
}
int main()
{
int n,k;
scanf("%d%d",&n,&k);
for (int i=1;i<=n;i++) scanf("%d",&val[i]);
int m;
scanf("%d",&m);
for (int i=0;i<m;i++) scanf("%d%d",&x[i],&y[i]);
for (int i=0;i<m;i++) G[x[i]].push_back(y[i]);
for (int i=1;i<=n;i++) tj[i][val[i]]++;
for (int i=1;i<=n;i++)
{
for (int j=1;j<=20;j++)
{
tj[i][j]+=tj[i-1][j];
}
}
for (int i=1;i<=n;i++)
{
if (G[i].size()>1) sort(G[i].begin(),G[i].end());
}
for (int i=0;i<m;i++) cou[x[i]]++;
for (int i=1;i<=n;i++) cou[i]=cou[i]+cou[i-1];
long long cnt=0;
for (int i=0;i<m;i++)
{
for (int j=0;j<G[x[i]].size();j++)
{
if (G[x[i]][j]==y[i]) break;
if (y[i]-G[x[i]][j]<=k) cnt++;
}
if (k>38)
{
if (x[i]>=k-38+1) cnt+=cou[x[i]-1]-cou[x[i]-(k-38)-1];
else cnt+=cou[x[i]-1];
}
for (int j=x[i]-k;j<min(x[i]-(k-38),x[i]);j++)
{
if (j<=0) continue;
int d=G[j].size();
int q=querymi(min(x[i],j),max(x[i],j));
for (int t=0;t<d;t++)
{
int dd;
if (q<y[i] && q<G[j][t]) dd=abs(x[i]-j)+y[i]-q+G[j][t]-q;
else dd=abs(x[i]-j)+abs(y[i]-G[j][t]);
if (dd<=k) cnt++;
}
}
}
printf("%lld\n",cnt);
return 0;
}
微軟大樓設計方案(困難)