Visible Trees HDU - 2841 （容斥原理）

Visible Trees

 HDU - 2841 

题意：  N*M的格点上有树（从1开始），从0，0点可以看到多少棵树。

gcd(x,y)！=1的格子看不到。

对每一列分别去求那些行的格子可以看到。

先预处理出1到m的素因子，然后利用容斥原理，对于行数是（偶数个素因子的积）的倍数的行，加上，否则减去。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=100010;
vector<int> pri[maxn];
int que[maxn];

void init(){
    for(int i=2;i<maxn;i++){
        pri[i].clear();
        int temp=i;
        for(int j=2;j*j<=i;j++) if(temp%j==0){
            pri[i].push_back(j);
            while(temp%j==0) temp/=j;
        }
        if(temp>1) pri[i].push_back(temp);
    }
}
ll solve(int n,int s){
    int cnt=0;
    que[cnt++]=1;
    for(int i=0;i<pri[s].size();i++){
        int v=pri[s][i];
        if(v>n) break;
        int k=cnt;
        for(int j=0;j<k;j++){
            que[cnt++]=que[j]*v*(-1);
        }
    }
    ll sum=0;
    for(int i=0;i<cnt;i++) sum+=n/que[i];
    return sum;
}

int main(){
    int t;
    init();
    scanf("%d",&t);
    while(t--){
        int n,m;
        scanf("%d%d",&n,&m);
        ll ans=n;
        for(int i=2;i<=m;i++) ans+=solve(n,i);
        printf("%lld\n",ans);
    }
}