鏈接：https://ac.nowcoder.com/acm/contest/894/B?&headNav=acm
來源：牛客網

華華送奕奕小禮物
時間限制：C/C++ 1秒，其他語言2秒
空間限制：C/C++ 524288K，其他語言1048576K
64bit IO Format: %lld
題目描述
華華用數組a和數組b合成了矩陣c。其中a數組長度為n，b數組長度為m，c是n行m列的矩陣，且c[i][j]=a[i]*b[j]。定義矩陣的權值為矩陣中所有元素的和。然后他想把矩陣送給奕奕。然而他怕奕奕不喜歡。若矩陣的權值小于L,奕奕會討厭它，因為奕奕不喜歡太小的數字。若矩陣的權值大于R，奕奕會生氣因為奕奕不認識比R大的數字。所以奕奕只喜歡權值大于等于L并且小于等于R的矩陣。還好華華學過acm，他馬上想到可以送奕奕一個子矩陣，并且他立馬寫程序從c矩陣中找出了所有奕奕喜歡的子矩陣。你只需要幫他算算這樣的子矩陣有多少個即可。
輸入描述:
第一行輸入n,m,L,R。
第二行n個數表示a數組
第三行m個數表示b數組
1<=n,m<=1000,1<=L<=R<=1e18
1<=a[i],b[i]<=1e6
輸出描述:
輸出一個數表示子矩陣的個數
示例1
輸入
復制
3 3 3 8
3 2 3
2 3 1
輸出
復制
10

思路：

我們定義 suma,sumb 分別是a，b數組的前綴和。

那么c矩陣中，左上角是i，j，右下角是x，y的sum和就是 (suma[x]-suma[i-1] ) * (sumb[y] - sumb[j-1] )

那么我們可以預處理出 (sumb[y] - sumb[j-1] ) ，加入一個數組v中并且排序，然后枚舉 (suma[x]-suma[i-1] ) ，去二分查找v中使其乘積滿足l~r的范圍。

答案加起來即可。

細節見代碼：

#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #include <cmath> #include <queue> #include <stack> #include <map> #include <set> #include <vector> #include <iomanip> #define ALL(x) (x).begin(), (x).end() #define sz(a) int(a.size()) #define all(a) a.begin(), a.end() #define rep(i,x,n) for(int i=x;i<n;i++) #define repd(i,x,n) for(int i=x;i<=n;i++) #define pii pair<int,int> #define pll pair<long long ,long long> #define gbtb ios::sync_with_stdio(false),cin.tie(0),cout.tie(0) #define MS0(X) memset((X), 0, sizeof((X))) #define MSC0(X) memset((X), '\0', sizeof((X))) #define pb push_back #define mp make_pair #define fi first #define se second #define eps 1e-6 #define gg(x) getInt(&x) #define chu(x) cout<<"["<<#x<<" "<<(x)<<"]"<<endl using namespace std; typedef long long ll; ll gcd(ll a,ll b){return b?gcd(b,a%b):a;} ll lcm(ll a,ll b){return a/gcd(a,b)*b;} ll powmod(ll a,ll b,ll MOD){ll ans=1;while(b){if(b%2)ans=ans*a%MOD;a=a*a%MOD;b/=2;}return ans;} inline void getInt(int* p); const int maxn=1010; const int inf=0x3f3f3f3f; /*** TEMPLATE CODE * * STARTS HERE ***/int n,m; ll l,r; ll a[maxn]; ll b[maxn]; ll suma[maxn]; ll sumb[maxn]; std::vector<ll> v; int main() {//freopen("D:\\common_text\\code_stream\\in.txt","r",stdin);//freopen("D:\\common_text\\code_stream\\out.txt","w",stdout);gbtb;cin>>n>>m;cin>>l>>r;repd(i,1,n){cin>>a[i];suma[i]=suma[i-1]+a[i];}repd(i,1,m){cin>>b[i];sumb[i]=sumb[i-1]+b[i];}repd(i,1,m){repd(j,1,i){v.push_back(sumb[i]-sumb[j-1]);}}ll ans=0ll;sort(ALL(v));repd(i,1,n){repd(j,1,i){ll u=r/(suma[i]-suma[j-1]);ll d=ceil(l*1.00/(suma[i]-suma[j-1]));ans+=upper_bound(ALL(v),u)-lower_bound(ALL(v),d);}}cout<<ans<<endl;return 0; }inline void getInt(int* p) {char ch;do {ch = getchar();} while (ch == ' ' || ch == '\n');if (ch == '-') {*p = -(getchar() - '0');while ((ch = getchar()) >= '0' && ch <= '9') {*p = *p * 10 - ch + '0';}}else {*p = ch - '0';while ((ch = getchar()) >= '0' && ch <= '9') {*p = *p * 10 + ch - '0';}} }

轉載于:https://www.cnblogs.com/qieqiemin/p/11343425.html

總結

以上是生活随笔為你收集整理的牛客练习赛46 B华华送奕奕小礼物 （预处理前缀和，二分）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。