题目描述
小朋友排成一排,老师给他们分苹果。
小朋友从左到右标号1..N。有M个老师,每次第i个老师会给第Li个到第Ri个,一共Ri-Li+1个小朋友每人发Ci个苹果。
最后老师想知道每个小朋友有多少苹果。
数据规模和约定
100%的数据,N、M≤100 000,1≤Li≤Ri≤N,0≤Ci≤100。
输入
第一行两个整数N、M,表示小朋友个数和老师个数。
接下来M行,每行三个整数Li、Ri、Ci,意义如题目表述。
输出
一行N个数,第i个数表示第i个小朋友手上的水果。
样例输入
5 3
1 2 1
2 3 2
2 5 3
样例输出
1 6 5 3 3
超时间了,想用离散化弄一下,但是好像是使用线段树。
#include<iostream>
#include<vector> #include<algorithm> using namespace std; typedef long long LL; const int maxn = 100001; vector<int>bound; int c[maxn],L[maxn],R[maxn]; int n,m; LL res[maxn]; int main(void) { cin >> n >> m; for(int i=1;i<=m;i++) { cin >> L[i] >> R[i] >> c[i]; bound.push_back(L[i]); bound.push_back(R[i]); if(L[i]>1) bound.push_back(L[i]-1); if(R[i]<n) bound.push_back(R[i]+1); } bound.push_back(1); bound.push_back(n); sort(bound.begin(),bound.end()); bound.erase(unique(bound.begin(),bound.end()),bound.end()); for(int i=0;i<bound.size();i++) cout << bound[i]<<" "; cout << endl; for(int i=1;i<=m;i++) { int s=find(bound.begin(),bound.end(),L[i])-bound.begin(); int e=find(bound.begin(),bound.end(),R[i])-bound.begin(); for(int j=s;j<=e;j++) res[j]+=c[i]; } int cnt = bound.size(); for(int i=0;i+1<cnt;i++) { int num = bound[i+1]-bound[i]; if(i+1==cnt-1) num++; for(int j=0;j<num;j++) cout << res[i] << " "; } return 0; }
使用差分数组,感觉和线段树有点相似
#include<iostream>
#include<algorithm> using namespace std; const int maxn = 100001; int A[maxn],d[maxn]; int main(void) { int n,m; cin >> n >> m; for(int i=1;i<=m;i++) { int L,R,C; cin >> L >> R >> C; d[L]+=C; d[R+1]-=C; } for(int i=1;i<=n;i++) { if(i==1) { A[i]=d[i]; cout << A[i]; } else { A[i]=A[i-1]+d[i]; cout << A[i]; } if(i!=n) cout << " "; } return 0; }