hdu 4293 dp求最大权值不重合区间

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int maxn = 506;

struct Node{
int A,B;
int num;
bool operator < (const Node& rhs) const{
return B > rhs.B;
}
}group[maxn];

int G[maxn][maxn];
int dp[maxn];   //dp[i]表示前i个人，最大有多少个人没撒谎。

int main()
{
//freopen("E:\\acm\\input.txt","r",stdin);
int N;
while(cin>>N){
memset(G,0,sizeof(G));
for(int i=1;i<=N;i++){
int a,b;
scanf("%d %d",&a,&b);
if(a+b<N && G[a][b]+a+b<N)  //这句才是关键。WA的原因。
G[a][b]++;
}
int cnt = 0;
for(int i=0;i<N;i++)
for(int j=0;j<N;j++){
if(G[i][j]){
group[cnt].A = i;
group[cnt].B = j;
group[cnt++].num = G[i][j];
}
}
sort(group,group+cnt);
memset(dp,0,sizeof(dp));
int pv = 0;
int i;
for(i=1;i<=N;i++){
if(pv >= cnt)  break;
int a = group[pv].A;
int num = group[pv].num;
int b = group[pv].B;
if(N-b == i){
dp[i] = max(dp[i-1],dp[a]+num);
pv++;
while(pv < cnt && N-group[pv].B == i){
dp[i] = max(dp[i],dp[group[pv].A]+group[pv].num);
pv++;
}
}
else{
dp[i] = dp[i-1];
}
}
printf("%d\n",dp[i-1]);
}
}