題干：

小a有n個(gè)數(shù)，他想把他們劃分為連續(xù)的權(quán)值相等的k段，但他不知道這是否可行。

每個(gè)數(shù)都必須被劃分

這個(gè)問題對他來說太難了，于是他把這個(gè)問題丟給了你。

輸入描述:

第一行為兩個(gè)整數(shù)n,q，分別表示序列長度和詢問個(gè)數(shù)。 第二行有n個(gè)數(shù)，表示序列中的每個(gè)數(shù)。 接下來的q行，每行包含一個(gè)數(shù)k，含義如題所示。

輸出描述:

輸出q行，每行對應(yīng)一個(gè)數(shù)Yes或者No，分別表示可行/不可行

?

示例1

輸入

復(fù)制

5 3 2 1 3 -1 4 3 2 1

輸出

復(fù)制

Yes No Yes

備注:

?

對于的數(shù)據(jù)，

對于的數(shù)據(jù)，

對于的數(shù)據(jù)，

設(shè)ai表示數(shù)列中的第i個(gè)數(shù)，保證

保證數(shù)據(jù)完全隨機(jī)

?

解題報(bào)告：

? ?剛開始以為二分復(fù)雜度是正確的，寫了一個(gè)就AC了，但是后來一想發(fā)現(xiàn)不對啊復(fù)雜度成了O(q^2logn)、、、大概是數(shù)據(jù)水了吧

? ?而且我這種解法根本不適合有負(fù)數(shù)存在的情況、、因?yàn)閟um數(shù)組就不單調(diào)了呀。

AC代碼：

#include<bits/stdc++.h> #define ll long long using namespace std; const int MAX = 100000 + 5 ; ll a[MAX]; ll sum[MAX]; int main() {int n,q,k;cin>>n>>q;for(int i = 1; i<=n; i++) scanf("%lld",a+i),sum[i] = sum[i-1] + a[i];while(q--) {scanf("%d",&k);if(sum[n] % k != 0) {puts("No");continue;}int every = sum[n] / k;int cur = 0,flag = 1;for(int i = 1; i<=k; i++) {int pos = lower_bound(sum+1,sum+n+1,cur + every) - sum;if(sum[pos] != cur+every) {flag=0;break;}cur += every;}if(flag == 1) puts("Yes");else puts("No");}// 2 3 6 5 9return 0 ;}

標(biāo)程：（其實(shí)也差不多啦復(fù)雜度o(因子個(gè)數(shù)*N + q)）

? ? 其實(shí)就是打表算的，對于這題其實(shí)打表比較合適，因?yàn)閝比n大，且題干中說了數(shù)據(jù)保證隨機(jī)，所以最好是打表然后o(1)查詢

#include<cstdio> #include<cstring> #include<algorithm> #define LL long long using namespace std; const int MAXN = 2 * 1e6 + 10, INF = 1e9 + 10; inline int read() {char c = getchar(); int x = 0, f = 1;while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();return x * f; } int a[MAXN]; bool ans[MAXN]; int main() {int N = read(), Q = read();LL sum = 0;for(int i = 1; i <= N; i++) a[i] = read(), sum += a[i];for(int i = 1; i <= N; i++) {if(sum % i != 0) {ans[i] = 0; continue;}LL cur = 0, k = 0;for(int j = 1; j <= N; j++) {cur += a[j];if(cur == sum / i) cur = 0, k++;}ans[i] = (cur == 0 && k == i);}while(Q--) {int x = read();puts(!ans[x] ? "No" : "Yes");} }

數(shù)據(jù)保證隨機(jī)的意思是 sum 的因子不會(huì)太多（構(gòu)造數(shù)據(jù)可以達(dá)到1e5級(jí)別）
另外可能有一個(gè)坑點(diǎn)：因?yàn)橛胸?fù)數(shù)的存在，如果當(dāng)前數(shù)大于了 sum/k 了，是不能直接跳出的（這是針對標(biāo)程的解法的，用前綴和就不存在這個(gè)問題）

還是要注意一下負(fù)數(shù)啊！！各種題中，尤其是那種，說 int范圍的。比如這題

---

不對啊，我那種方法其實(shí)修改一下也是正確的，用set維護(hù)一個(gè)pair<前綴，當(dāng)前下標(biāo)>，然后每次二分查找pair<那個(gè)值，上一次查找的下標(biāo)>，這樣找到的就是pair<那個(gè)值，那個(gè)下標(biāo)后面的值>或者pair<大于那個(gè)值，下標(biāo)無所謂>，我們在if判斷一下是否是第一種，就可以了。

總結(jié)

以上是生活随笔為你收集整理的【牛客 - 181C】序列（前缀和，二分，可用set维护）（有坑）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。